| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | bfc_allocator.cc | 228 const AllocationAttributes& allocation_attr) { in AllocateRawInternalWithRetry() argument
231 if (allocation_attr.freed_by_func != nullptr) { in AllocateRawInternalWithRetry()
232 freed_by_count = (*allocation_attr.freed_by_func)(); in AllocateRawInternalWithRetry()
241 [this, &allocation_attr](size_t a, size_t nb, bool v) { in AllocateRawInternalWithRetry()
243 if (allocation_attr.freed_by_func != nullptr) { in AllocateRawInternalWithRetry()
244 freed_by_count = (*allocation_attr.freed_by_func)(); in AllocateRawInternalWithRetry()
254 const AllocationAttributes& allocation_attr) { in AllocateRaw() argument
256 if (!allocation_attr.retry_on_failure) { in AllocateRaw()
261 if (allocation_attr.freed_by_func != nullptr) { in AllocateRaw()
262 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.retry_on_failure = false; in AllocateBytes()
74 AllocatorAttributes(), allocation_attr)); in AllocateBytes()
|
| D | fft_ops.cc | 374 AllocationAttributes allocation_attr; in AllocateBytes() local
375 allocation_attr.retry_on_failure = false; in AllocateBytes()
378 AllocatorAttributes(), allocation_attr)); in AllocateBytes()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | typed_allocator.h | 38 const AllocationAttributes& allocation_attr) { in Allocate() argument
48 sizeof(T) * num_elements, allocation_attr); in Allocate()
|
| D | allocator.h | 202 const AllocationAttributes& allocation_attr) { in AllocateRaw() argument
311 const AllocationAttributes& allocation_attr) override { in AllocateRaw() argument
312 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 | 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()
778 const AllocationAttributes& allocation_attr) in Tensor() argument
783 CASES(type, buf_ = new Buffer<T>(a, shape.num_elements(), allocation_attr)); in Tensor()
785 if (MemoryLoggingEnabled() && !allocation_attr.allocation_will_be_logged && in Tensor()
|
| D | op_kernel.cc | 708 AllocatorAttributes attr, const AllocationAttributes& allocation_attr) { in allocate_tensor() argument
713 /*retry_on_failure=*/allocation_attr.retry_on_failure, in allocate_tensor()
714 /*allocation_will_be_logged=*/true, allocation_attr.freed_by_func)); in allocate_tensor()
778 const AllocationAttributes& allocation_attr) { in allocate_temp() argument
798 allocate_tensor(type, shape, out_temp, allocator_attr, allocation_attr); in allocate_temp()
|
| D | op_kernel.h | 1013 const AllocationAttributes& allocation_attr);
1285 const AllocationAttributes& allocation_attr);
|
| D | tensor.h | 153 const AllocationAttributes& allocation_attr);
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | tensor_coding.h | 80 const AllocationAttributes& allocation_attr);
|
| D | base_rendezvous_mgr.cc | 268 AllocationAttributes allocation_attr; in SameWorkerRecvDone() local
276 allocation_attr.freed_by_func = &freed_by_func; in SameWorkerRecvDone()
280 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 | 648 AllocationAttributes allocation_attr; in MaybeCopyTensorToGPU() local
655 allocation_attr.freed_by_func = &freed_by_func; in MaybeCopyTensorToGPU()
658 from.shape(), allocation_attr); in MaybeCopyTensorToGPU()
|