Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 9 of 9 for reference (0.21 sec)

  1. 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)
  2. tensorflow/c/c_api_experimental.cc

      using tensorflow::NodeDef;
      using tensorflow::OpRegistrationData;
      using tensorflow::Tensor;
      using tensorflow::shape_inference::DimensionHandle;
      using tensorflow::shape_inference::InferenceContext;
      using tensorflow::shape_inference::ShapeAndType;
      using tensorflow::shape_inference::ShapeHandle;
    
      const int num_inputs = input_shapes->num_items;
      NodeDef node_def;
    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/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/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)
  5. tensorflow/c/experimental/grappler/grappler.cc

          graph_properties);
    }
    
    void TF_InferStatically(TF_GraphProperties* graph_properties,
                            TF_Bool assume_valid_feeds,
                            TF_Bool aggressive_shape_inference,
                            TF_Bool include_input_tensor_values,
                            TF_Bool include_output_tensor_values,
                            TF_Status* status) {
      TF_SetStatus(status, TF_OK, "");
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Wed Sep 06 19:12:29 GMT 2023
    - 15K bytes
    - Viewed (1)
  6. tensorflow/c/eager/parallel_device/parallel_device.cc

      return result;
    }
    
    // Used as an argument to TFE_NewCustomDeviceTensorHandle, indicating how
    // ParallelTensors wrapped in TFE_TensorHandles should be cleaned up once their
    // reference counts drop to zero.
    void ParallelTensorDeallocator(void* data) {
      delete reinterpret_cast<ParallelTensor*>(data);
    }
    
    // Used as an argument to TFE_NewCustomDeviceTensorHandle, for computing the
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 29 22:05:31 GMT 2023
    - 18.3K bytes
    - Viewed (0)
  7. 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)
  8. tensorflow/c/eager/c_api_unified_experimental_graph.cc

    namespace tensorflow {
    namespace tracing {
    namespace graph {
    
    class GraphContext;
    class GraphOperation;
    class GraphTensor;
    
    auto& kUnknownDim = shape_inference::InferenceContext::kUnknownDim;
    auto& kUnknownRank = shape_inference::InferenceContext::kUnknownRank;
    
    // GraphTensor wraps a `TF_Output`, i.e. a pointer to TF_Operation and the index
    // into the list of outputs for the operation.
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 15.4K bytes
    - Viewed (1)
  9. tensorflow/c/c_api_experimental_test.cc

      }
    
      ~ShapeInferenceTest() override {
        TFE_DeleteContextOptions(tfe_context_options_);
        TFE_DeleteContext(tfe_context_);
        TF_DeleteStatus(status_);
      }
    
      // Checks the expected result of shape inference for the given `op`.
      void CheckOutputShapes(
          TFE_Op* op,
          const std::vector<absl::optional<std::vector<int64_t>>>& input_shapes_vec,
          const std::vector<TF_Tensor*>& input_tensors,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Jan 17 22:27:52 GMT 2023
    - 13.1K bytes
    - Viewed (1)
Back to top