Searched refs:allocate_tensor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.cc | 702 Status OpKernelContext::allocate_tensor( in allocate_tensor() function in tensorflow::OpKernelContext
734 Status s = allocate_tensor(type, shape, output_tensor.get(), attr); in allocate_output()
747 allocate_tensor(type, shape, out_temp, allocator_attr, allocation_attr); in allocate_temp()
764 Status s = allocate_tensor(type, shape, &persistent, attr); in allocate_persistent()
|
| D | op_kernel.h | 1239 Status allocate_tensor(DataType type, const TensorShape& shape, in allocate_tensor() function
1242 return allocate_tensor(type, shape, out_tensor, allocator_attr, in allocate_tensor()
1246 Status allocate_tensor(DataType type, const TensorShape& shape,
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 172 void* allocate_tensor(const char* operation, size_t len) { in allocate_tensor() function
199 void* data = allocate_tensor("TF_AllocateTensor", len); in TF_AllocateTensor()
225 buf = new TF_ManagedBuffer(allocate_tensor("TF_NewTensor", len), len, in TF_NewTensor()
|