| /external/tensorflow/tensorflow/core/framework/ |
| D | allocator.h | 143 const AllocationAttributes& allocation_attr) { in AllocateRaw() argument
252 const AllocationAttributes& allocation_attr) override { in AllocateRaw() argument
253 return wrapped_->AllocateRaw(alignment, num_bytes, allocation_attr); in AllocateRaw()
|
| D | typed_allocator.h | 38 const AllocationAttributes& allocation_attr) { in Allocate() argument
48 sizeof(T) * num_elements, allocation_attr); in Allocate()
|
| 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 | tracking_allocator.h | 62 const AllocationAttributes& allocation_attr) override;
|
| D | tensor.cc | 131 Buffer(Allocator* a, int64 n, const AllocationAttributes& allocation_attr);
477 const AllocationAttributes& allocation_attr) in Buffer() argument
478 : BufferBase(a, TypedAllocator::Allocate<T>(a, n, allocation_attr)), in Buffer()
785 const AllocationAttributes& allocation_attr) in Tensor() argument
790 CASES(type, buf_ = new Buffer<T>(a, shape.num_elements(), allocation_attr)); in Tensor()
792 if (MemoryLoggingEnabled() && !allocation_attr.allocation_will_be_logged && in Tensor()
|
| D | op_kernel.cc | 749 AllocatorAttributes attr, const AllocationAttributes& allocation_attr) { in allocate_tensor() argument
754 AllocationAttributes(allocation_attr.no_retry_on_failure, in allocate_tensor()
756 allocation_attr.freed_by_func)); in allocate_tensor()
819 const AllocationAttributes& allocation_attr) { in allocate_temp() argument
837 allocate_tensor(type, shape, out_temp, allocator_attr, allocation_attr); in allocate_temp()
|
| D | op_kernel.h | 1022 const AllocationAttributes& allocation_attr);
1312 const AllocationAttributes& allocation_attr);
|
| D | tensor.h | 150 const AllocationAttributes& allocation_attr);
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | bfc_allocator.cc | 202 const AllocationAttributes& allocation_attr) { in AllocateRawInternalWithRetry() argument
205 if (allocation_attr.freed_by_func != nullptr) { in AllocateRawInternalWithRetry()
206 freed_by_count = (*allocation_attr.freed_by_func)(); in AllocateRawInternalWithRetry()
215 [this, &allocation_attr](size_t a, size_t nb, bool v) { in AllocateRawInternalWithRetry()
217 if (allocation_attr.freed_by_func != nullptr) { in AllocateRawInternalWithRetry()
218 freed_by_count = (*allocation_attr.freed_by_func)(); in AllocateRawInternalWithRetry()
228 const AllocationAttributes& allocation_attr) { in AllocateRaw() argument
230 if (allocation_attr.no_retry_on_failure) { in AllocateRaw()
235 if (allocation_attr.freed_by_func != nullptr) { in AllocateRaw()
236 freed_by_count = (*allocation_attr.freed_by_func)(); in AllocateRaw()
[all …]
|
| D | bfc_allocator.h | 64 const AllocationAttributes& allocation_attr) override;
97 const AllocationAttributes& allocation_attr);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_gpu.h | 70 AllocationAttributes allocation_attr; in AllocateBytes() local
71 allocation_attr.no_retry_on_failure = true; in AllocateBytes()
74 AllocatorAttributes(), allocation_attr)); in AllocateBytes()
|
| D | fft_ops.cc | 374 AllocationAttributes allocation_attr; in AllocateBytes() local
375 allocation_attr.no_retry_on_failure = true; in AllocateBytes()
378 AllocatorAttributes(), allocation_attr)); in AllocateBytes()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | tensor_coding.h | 80 const AllocationAttributes& allocation_attr);
|
| D | base_rendezvous_mgr.cc | 283 AllocationAttributes allocation_attr; in SameWorkerRecvDone() local
291 allocation_attr.freed_by_func = &freed_by_func; in SameWorkerRecvDone()
295 Tensor copy(out_allocator, in.dtype(), in.shape(), allocation_attr); in SameWorkerRecvDone()
|
| D | tensor_coding.cc | 72 const AllocationAttributes& allocation_attr) { in InitPartial() argument
78 Tensor t(allocator_, meta_.tensor().dtype(), shape, allocation_attr); in InitPartial()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 591 AllocationAttributes allocation_attr; in MaybeCopyTensorToGPU() local
598 allocation_attr.freed_by_func = &freed_by_func; in MaybeCopyTensorToGPU()
601 from.shape(), allocation_attr); in MaybeCopyTensorToGPU()
|