Searched refs:allocator_attr (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 871 const TensorShape& shape, const AllocatorAttributes& allocator_attr,
968 Tensor* out_temp, AllocatorAttributes allocator_attr,
971 Tensor* out_temp, AllocatorAttributes allocator_attr) { in allocate_temp() argument
972 return allocate_temp(type, shape, out_temp, allocator_attr, in allocate_temp()
1241 AllocatorAttributes allocator_attr) { in allocate_tensor() argument
1242 return allocate_tensor(type, shape, out_tensor, allocator_attr, in allocate_tensor()
1247 Tensor* out_tensor, AllocatorAttributes allocator_attr,
|
| D | op_kernel.cc | 566 const TensorShape& shape, const AllocatorAttributes& allocator_attr, in forward_input_or_allocate_temp() argument
571 type, shape, DEVICE_MEMORY, allocator_attr); in forward_input_or_allocate_temp()
577 return allocate_temp(type, shape, out_temp, allocator_attr); in forward_input_or_allocate_temp()
744 AllocatorAttributes allocator_attr, in allocate_temp() argument
747 allocate_tensor(type, shape, out_temp, allocator_attr, allocation_attr); in allocate_temp()
749 Allocator* a = get_allocator(allocator_attr); in allocate_temp()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | scoped_allocator.h | 102 const AllocationAttributes& allocator_attr) override { in AllocateRaw() argument
|