- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for MakeTensor (0.08 sec)
-
tensorflow/c/tf_tensor_internal.h
void* const deallocator_arg_; bool owns_memory_; }; namespace tensorflow { class TensorCApi { public: static TensorBuffer* Buffer(const Tensor& tensor) { return tensor.buf_; } static Tensor MakeTensor(TF_DataType type, const TensorShape& shape, TensorBuffer* buf) { return Tensor(static_cast<DataType>(type), shape, buf); } }; // Allocates tensor data buffer using specified allocator.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 24 20:38:55 UTC 2023 - 4.6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util.cc
xla_allocator_); } return std::move(arguments); } // Construct the tensor for the given type and buffer. static Tensor MakeTensor(DataType dtype, const TensorShape& shape, se::DeviceMemoryBase buffer, Allocator* allocator) { size_t expected_size = shape.num_elements() * DataTypeSize(dtype);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 40.4K bytes - Viewed (0)