Searched refs:allocator_attr (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.cc | 282 AllocatorAttributes allocator_attr) { in allocate_temp() argument
283 if (allocator_attr.scope_id != 0) { in allocate_temp()
287 Allocator* a = device_->GetAllocator(allocator_attr); in allocate_temp()
568 const TensorShape& shape, const AllocatorAttributes& allocator_attr, in forward_input_or_allocate_temp() argument
573 type, shape, DEVICE_MEMORY, allocator_attr); in forward_input_or_allocate_temp()
579 return allocate_temp(type, shape, out_temp, allocator_attr); in forward_input_or_allocate_temp()
777 AllocatorAttributes allocator_attr, in allocate_temp() argument
779 if (allocator_attr.scope_id > 0) { in allocate_temp()
791 << " called allocate_temp with scope_id " << allocator_attr.scope_id in allocate_temp()
793 allocator_attr.scope_id = -1; in allocate_temp()
[all …]
|
| D | op_kernel.h | 289 Tensor* out_temp, AllocatorAttributes allocator_attr);
916 const TensorShape& shape, const AllocatorAttributes& allocator_attr,
1012 Tensor* out_temp, AllocatorAttributes allocator_attr,
1015 Tensor* out_temp, AllocatorAttributes allocator_attr) { in allocate_temp() argument
1016 return allocate_temp(type, shape, out_temp, allocator_attr, in allocate_temp()
1278 AllocatorAttributes allocator_attr) { in allocate_tensor() argument
1279 return allocate_tensor(type, shape, out_tensor, allocator_attr, in allocate_tensor()
1284 Tensor* out_tensor, AllocatorAttributes allocator_attr,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | scoped_allocator.h | 105 const AllocationAttributes& allocator_attr) override { in AllocateRaw() argument
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels.cc | 530 tensorflow::AllocatorAttributes allocator_attr; in TF_AllocateTemp() local
532 allocator_attr.set_on_host(true); in TF_AllocateTemp()
538 allocator_attr); in TF_AllocateTemp()
|