Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for references (0.69 sec)

  1. tensorflow/c/eager/gradients.cc

      ~TapeVSpace() override {}
    
      // Returns the number of elements in the gradient tensor.
      int64_t NumElements(AbstractTensorHandle* tensor) const override;
    
      // Consumes references to the tensors in the gradient_tensors list and returns
      // a tensor with the result.
      AbstractTensorHandle* AggregateGradients(
          gtl::ArraySlice<AbstractTensorHandle*> gradient_tensors) const override;
    
    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)
  2. tensorflow/c/eager/dlpack.cc

    // original framework of destruction, and this context will be deleted also.
    struct TfDlManagedTensorCtx {
      TensorReference reference;
      std::vector<int64_t> shape;
      std::vector<int64_t> strides;
      DLManagedTensor tensor;
    
      explicit TfDlManagedTensorCtx(const TensorReference& ref) : reference(ref) {}
    };
    
    // Gets tensor from eager tensor handle.
    const Tensor* GetTensorFromHandle(TFE_TensorHandle* h, TF_Status* status) {
    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)
  3. tensorflow/c/c_api_function.cc

          const auto& iter = input_nodes.find(node);
          if (iter == input_nodes.end()) {
            // This node is not referenced in inputs. Add it to the body.
            body_nodes->push_back(node);
          } else {
            // This node is referenced in inputs. Currently, we place an
            // artificial restriction and require that when num_opers=-1, such
            // nodes must have a single output.
    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)
  4. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

        std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> await_status(
            TF_NewStatus(), TF_DeleteStatus);
        // Wait until all pending nodes have completed since they may have a
        // reference to default_cancellation_manager_. We ignore the status return
        // since we already have a bad status to propagate.
        TFE_ContextAsyncWait(context, await_status.get());
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Feb 09 07:47:20 GMT 2024
    - 25.4K bytes
    - Viewed (1)
Back to top