Searched refs:allocation_attr (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | allocator.h | 94 const AllocationAttributes& allocation_attr) { in AllocateRaw() argument
116 const AllocationAttributes& allocation_attr) { in Allocate() argument
125 allocation_attr); in Allocate()
301 const AllocationAttributes& allocation_attr) override { in AllocateRaw() argument
302 return wrapped_->AllocateRaw(alignment, num_bytes, allocation_attr); in AllocateRaw()
|
| D | tracking_allocator.cc | 34 const AllocationAttributes& allocation_attr) { in AllocateRaw() argument
35 void* ptr = allocator_->AllocateRaw(alignment, num_bytes, allocation_attr); in AllocateRaw()
|
| D | tensor.cc | 104 Buffer(Allocator* a, int64 n, const AllocationAttributes& allocation_attr);
442 const AllocationAttributes& allocation_attr) in Buffer() argument
443 : BufferBase(a), data_(a->Allocate<T>(n, allocation_attr)), elem_(n) {} in Buffer()
727 const AllocationAttributes& allocation_attr) in Tensor() argument
732 CASES(type, buf_ = new Buffer<T>(a, shape.num_elements(), allocation_attr)); in Tensor()
734 if (!allocation_attr.allocation_will_be_logged && buf_ != nullptr && in Tensor()
|
| D | tracking_allocator.h | 64 const AllocationAttributes& allocation_attr) override;
|
| D | op_kernel.cc | 628 AllocatorAttributes attr, const AllocationAttributes& allocation_attr) { in allocate_tensor() argument
630 AllocationAttributes logged_attr(allocation_attr); in allocate_tensor()
669 const AllocationAttributes& allocation_attr) { in allocate_temp() argument
671 allocate_tensor(type, shape, out_temp, allocator_attr, allocation_attr); in allocate_temp()
|
| D | op_kernel.h | 833 const AllocationAttributes& allocation_attr);
1093 const AllocationAttributes& allocation_attr);
|
| D | tensor.h | 102 const AllocationAttributes& allocation_attr);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_gpu.h | 59 AllocationAttributes allocation_attr; in AllocateBytes() local
60 allocation_attr.no_retry_on_failure = true; in AllocateBytes()
63 AllocatorAttributes(), allocation_attr)); in AllocateBytes()
|
| D | fft_ops.cc | 316 AllocationAttributes allocation_attr; in AllocateBytes() local
317 allocation_attr.no_retry_on_failure = true; in AllocateBytes()
320 AllocatorAttributes(), allocation_attr)); in AllocateBytes()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | bfc_allocator.cc | 207 const AllocationAttributes& allocation_attr) { in AllocateRaw() argument
208 if (allocation_attr.no_retry_on_failure) { in AllocateRaw()
|
| D | bfc_allocator.h | 55 const AllocationAttributes& allocation_attr) override;
|