Searched refs:allocate_tensor (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/c/ |
| D | tf_tensor.cc | 39 void* allocate_tensor(const char* operation, size_t len, Allocator* allocator) { in allocate_tensor() function
49 void* allocate_tensor(const char* operation, size_t len) { in allocate_tensor() function
50 return allocate_tensor(operation, len, cpu_allocator()); in allocate_tensor()
92 void* data = tensorflow::allocate_tensor("TF_AllocateTensor", len, in TF_AllocateTensor()
117 buf = new TF_ManagedBuffer(tensorflow::allocate_tensor("TF_NewTensor", len), in TF_NewTensor()
|
| D | tf_tensor_internal.h | 85 void* allocate_tensor(const char* operation, size_t len, Allocator* allocator);
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | LogicalFixture.h | 59 void allocate_tensor(std::initializer_list<TensorType *> tensors) in allocate_tensor() function
100 Parent::allocate_tensor({ &src0, &src1, &dst }); in compute_target()
155 Parent::allocate_tensor({ &src, &dst }); in compute_target()
|
| D | PixelWiseMultiplicationFixture.h | 103 auto allocate_tensor = [](TensorType & t) in compute_target()
114 allocate_tensor(src1); in compute_target()
115 allocate_tensor(src2); in compute_target()
120 allocate_tensor(dst); in compute_target()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 1237 Status allocate_tensor(DataType type, const TensorShape& shape, in allocate_tensor() function
1240 return allocate_tensor(type, shape, out_tensor, allocator_attr, in allocate_tensor()
1244 Status allocate_tensor(DataType type, const TensorShape& shape,
|
| D | op_kernel.cc | 750 Status OpKernelContext::allocate_tensor( in allocate_tensor() function in tensorflow::OpKernelContext
812 Status s = allocate_tensor(type, shape, output_tensor.get(), attr); in allocate_output()
843 allocate_tensor(type, shape, out_temp, allocator_attr, allocation_attr); in allocate_temp()
948 Status s = allocate_tensor(tensor.dtype(), tensor.shape(), new_tensor.get(), in maybe_set_output_by_allocate_and_copy()
|