Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 14 for Ehat (0.95 sec)

  1. tensorflow/c/c_api.cc

      return true;
    }
    
    // Copies `src_graph` into `dst_graph`. Any node in `src_graph` with input
    // `src_inputs[i]` will have that input replaced with `dst_inputs[i]`.  `prefix`
    // will be prepended to copied node names. `control_deps` are nodes in
    // `dst_graph` that the copied `src_graph` nodes will have control dependencies
    // on. `return_nodes` are nodes in `src_graph`, and the new corresponding nodes
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 102.3K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_experimental.cc

      // TODO(b/113217601): This is needed for EagerContext::runner_ to use a
      // threadpool, so that we avoid the possibility of running the runner_ in the
      // threadpool of GPU event mgr, as that can trigger more callbacks to be
      // scheduled on that same threadpool, causing a deadlock in cases where the
      // caller of event_mgr->ThenExecute() blocks on the completion of the callback
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
  3. tensorflow/c/experimental/gradients/array_grad.cc

        for (int i = 0; i < grad_outputs.size(); i++) {
          auto grad_input = grad_outputs[i];
          // TODO(srbs): Should we add a copy contructor to AbstractTensorHandle
          // that takes care of this similar to `Tensor`?
          if (grad_input) {
            grad_input->Ref();
          }
          grad_inputs[i] = grad_input;
        }
        return absl::OkStatus();
      }
    C++
    - Registered: Tue Apr 09 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 1.6K bytes
    - Viewed (0)
  4. tensorflow/c/experimental/filesystem/plugins/posix/posix_filesystem.cc

                           const char* dst, TF_Status* status) {
      // If target is a directory return TF_FAILED_PRECONDITION.
      // Target might be missing, so don't error in that case.
      struct stat st;
      if (stat(dst, &st) != 0) {
        if (errno != ENOENT) {
          TF_SetStatusFromIOError(status, errno, dst);
          return;
        }
      } else if (S_ISDIR(st.st_mode)) {
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Sun Mar 24 20:08:23 GMT 2024
    - 15.8K bytes
    - Viewed (0)
  5. tensorflow/c/eager/immediate_execution_tensor_handle.cc

    }
    
    Status ImmediateExecutionTensorHandle::SummarizeValue(
        std::string& summary) const {
      Status status;
      AbstractTensorPtr resolved(
          // TODO(allenl): Resolve should be const, and the caches that get updated
          // marked mutable.
          const_cast<ImmediateExecutionTensorHandle*>(this)->Resolve(&status));
      if (!status.ok()) {
        return status;
      }
      summary = resolved->SummarizeValue();
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 2.1K bytes
    - Viewed (0)
  6. tensorflow/c/eager/gradients.cc

      std::unordered_map<int64_t, TapeTensor> sources_that_are_targets;
      for (int i = 0; i < target_tensor_ids.size(); ++i) {
        int64_t target_id = target_tensor_ids[i];
        if (sources_set.find(target_id) != sources_set.end()) {
          auto tensor = targets[i];
          sources_that_are_targets.insert(
              std::make_pair(target_id, TapeTensor(tensor)));
        }
      }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  7. tensorflow/c/eager/dlpack.cc

      bool valid = true;
      int64_t expected_stride = 1;
      for (int i = ndim - 1; i >= 0; --i) {
        // Empty tensors are always compact regardless of strides.
        if (shape_arr[i] == 0) return true;
        // Note that dimensions with size=1 can have any stride.
        if (shape_arr[i] != 1 && stride_arr[i] != expected_stride) {
          valid = false;
        }
        expected_stride *= shape_arr[i];
      }
      return valid;
    }
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 12.8K bytes
    - Viewed (0)
  8. tensorflow/c/eager/c_api_distributed_test.cc

      TFE_TensorHandle* h0 = TestVariable(ctx, 1.0, task1_name);
      TFE_TensorHandle* h1 = TestVariable(ctx, 2.0, task2_name);
      TFE_TensorHandle* h2 = TestVariable(ctx, 3.0, task0_name);
    
      // Add a sync point to make sure that variables have been initialized
      // before the function execution starts.
      TFE_ContextAsyncWait(ctx, status);
      EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 23.5K bytes
    - Viewed (0)
  9. tensorflow/c/c_api_function.cc

                                        fn_name, "'");
        output_tensors->emplace_back(node, idx);
      }
      return absl::OkStatus();
    }
    
    // Populates `body_nodes` with the nodes that will become function's body.
    // Performs various checks.
    Status ComputeBodyNodes(
        const TF_Graph* fn_body, const char* fn_name, int num_opers,
        const TF_Operation* const* opers,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 13.6K bytes
    - Viewed (2)
  10. tensorflow/c/eager/unified_api_test.cc

      }
    
     public:
      bool UseMlir() const { return strcmp(std::get<0>(GetParam()), "mlir") == 0; }
      bool UseFunction() const { return std::get<2>(GetParam()); }
    };
    
    // Checks that inputs[0] is a scalar.
    Status TestScalarShape(AbstractContext* ctx,
                           absl::Span<AbstractTensorHandle* const> inputs,
                           absl::Span<AbstractTensorHandle*> outputs) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Feb 27 13:57:45 GMT 2024
    - 6.7K bytes
    - Viewed (0)
Back to top