- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 14 for reference (0.23 sec)
-
tensorflow/c/c_api.cc
} namespace { // Helper method that creates a shape handle for a shape described by dims. tensorflow::shape_inference::ShapeHandle ShapeHandleFromDims( tensorflow::shape_inference::InferenceContext* ic, int num_dims, const int64_t* dims) { if (num_dims != -1) { std::vector<tensorflow::shape_inference::DimensionHandle> dim_vec; dim_vec.reserve(num_dims); for (int i = 0; i < num_dims; ++i) {
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) -
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/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/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/experimental/filesystem/modular_filesystem_test.cc
s = fs->CreateDir(root_dir_); if (!s.ok()) { GTEST_SKIP() << "Cannot create working directory: " << s; } } // Converts path reference to URI reference. // // If URI scheme is empty, URI reference is `path` relative to current test // root directory. Otherwise, we need to add the `<scheme>://` in front of // this path. //
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri May 27 20:25:58 GMT 2022 - 71K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test.cc
EXPECT_EQ(orig_ptr, TF_TensorData(t)); } else { // In sync mode, forwarding can't really happen since the client code will // have a reference count on the input tensor while the kernel is being // executed and thus it cannot be re-used for the return tensor. EXPECT_NE(orig_ptr, TF_TensorData(t)); } }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 20:50:20 GMT 2023 - 94.6K bytes - Viewed (1) -
tensorflow/c/c_api_test.cc
.Attr("v: " #type) \ .SetShapeFn(tensorflow::shape_inference::UnknownShape); \ REGISTER_OP("CApiAttributesTestOpList" #type) \ .Attr("v: list(" #type ")") \ .SetShapeFn(tensorflow::shape_inference::UnknownShape) ATTR_TEST_REGISTER_OP(string); ATTR_TEST_REGISTER_OP(int); ATTR_TEST_REGISTER_OP(float);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 96.9K bytes - Viewed (3) -
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)