Searched refs:allocation_attr (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | bfc_allocator.cc | 187 const AllocationAttributes& allocation_attr) { in AllocateRawInternalWithRetry() argument
190 if (allocation_attr.freed_by_func != nullptr) { in AllocateRawInternalWithRetry()
191 freed_by_count = allocation_attr.freed_by_func(); in AllocateRawInternalWithRetry()
200 [this, &allocation_attr](size_t a, size_t nb, bool v) { in AllocateRawInternalWithRetry()
202 if (allocation_attr.freed_by_func != nullptr) { in AllocateRawInternalWithRetry()
203 freed_by_count = allocation_attr.freed_by_func(); in AllocateRawInternalWithRetry()
213 const AllocationAttributes& allocation_attr) { in AllocateRaw() argument
215 if (allocation_attr.no_retry_on_failure) { in AllocateRaw()
220 if (allocation_attr.freed_by_func != nullptr) { in AllocateRaw()
221 freed_by_count = allocation_attr.freed_by_func(); in AllocateRaw()
[all …]
|
| D | bfc_allocator.h | 60 const AllocationAttributes& allocation_attr) override;
87 const AllocationAttributes& allocation_attr);
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | allocator.h | 102 const AllocationAttributes& allocation_attr) { in AllocateRaw() argument
124 const AllocationAttributes& allocation_attr) { in Allocate() argument
133 allocation_attr); in Allocate()
305 const AllocationAttributes& allocation_attr) override { in AllocateRaw() argument
306 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 | 63 const AllocationAttributes& allocation_attr) override;
|
| D | tensor.cc | 109 Buffer(Allocator* a, int64 n, const AllocationAttributes& allocation_attr);
450 const AllocationAttributes& allocation_attr) in Buffer() argument
451 : BufferBase(a, a->Allocate<T>(n, allocation_attr)), elem_(n) {} in Buffer()
747 const AllocationAttributes& allocation_attr) in Tensor() argument
752 CASES(type, buf_ = new Buffer<T>(a, shape.num_elements(), allocation_attr)); in Tensor()
754 if (!allocation_attr.allocation_will_be_logged && buf_ != nullptr && in Tensor()
|
| D | op_kernel.cc | 704 AllocatorAttributes attr, const AllocationAttributes& allocation_attr) { in allocate_tensor() argument
706 AllocationAttributes logged_attr(allocation_attr); in allocate_tensor()
745 const AllocationAttributes& allocation_attr) { in allocate_temp() argument
747 allocate_tensor(type, shape, out_temp, allocator_attr, allocation_attr); in allocate_temp()
|
| D | op_kernel.h | 969 const AllocationAttributes& allocation_attr);
1248 const AllocationAttributes& allocation_attr);
|
| D | tensor.h | 108 const AllocationAttributes& allocation_attr);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_gpu.h | 57 AllocationAttributes allocation_attr; in AllocateBytes() local
58 allocation_attr.no_retry_on_failure = true; in AllocateBytes()
61 AllocatorAttributes(), allocation_attr)); in AllocateBytes()
|
| D | fft_ops.cc | 325 AllocationAttributes allocation_attr; in AllocateBytes() local
326 allocation_attr.no_retry_on_failure = true; in AllocateBytes()
329 AllocatorAttributes(), allocation_attr)); in AllocateBytes()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | tensor_coding.cc | 72 const AllocationAttributes& allocation_attr) { in InitPartial() argument
78 Tensor t(allocator_, meta_.tensor().dtype(), shape, allocation_attr); in InitPartial()
|
| D | tensor_coding.h | 80 const AllocationAttributes& allocation_attr);
|