- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for reference (0.21 sec)
-
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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)