Searched refs:AllocationAttributes (Results 1 – 24 of 24) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | allocator.h | 37 struct AllocationAttributes { struct
38 AllocationAttributes() = default;
40 AllocationAttributes(bool no_retry_on_failure, bool allocation_will_be_logged, in AllocationAttributes() function
62 TF_DISALLOW_COPY_AND_ASSIGN(AllocationAttributes); argument
143 const AllocationAttributes& allocation_attr) { in AllocateRaw()
252 const AllocationAttributes& allocation_attr) override { in AllocateRaw()
|
| D | tracking_allocator.h | 59 return AllocateRaw(alignment, num_bytes, AllocationAttributes()); in AllocateRaw()
62 const AllocationAttributes& allocation_attr) override;
|
| D | typed_allocator.h | 38 const AllocationAttributes& allocation_attr) { in Allocate()
|
| D | tracking_allocator.cc | 34 const AllocationAttributes& allocation_attr) { in AllocateRaw()
|
| D | op_kernel.h | 1022 const AllocationAttributes& allocation_attr);
1026 AllocationAttributes()); in allocate_temp()
1307 AllocationAttributes()); in allocate_tensor()
1312 const AllocationAttributes& allocation_attr);
|
| D | op_kernel.cc | 250 AllocationAttributes attr; in allocate_temp()
275 AllocationAttributes attr; in allocate_temp()
749 AllocatorAttributes attr, const AllocationAttributes& allocation_attr) { in allocate_tensor()
754 AllocationAttributes(allocation_attr.no_retry_on_failure, in allocate_tensor()
819 const AllocationAttributes& allocation_attr) { in allocate_temp()
|
| D | tensor.cc | 131 Buffer(Allocator* a, int64 n, const AllocationAttributes& allocation_attr);
472 : BufferBase(a, TypedAllocator::Allocate<T>(a, n, AllocationAttributes())), in Buffer()
477 const AllocationAttributes& allocation_attr) in Buffer()
785 const AllocationAttributes& allocation_attr) in Tensor()
|
| D | tensor.h | 150 const AllocationAttributes& allocation_attr);
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | tensor_coding.h | 80 const AllocationAttributes& allocation_attr);
|
| D | tensor_coding.cc | 72 const AllocationAttributes& allocation_attr) { in InitPartial()
|
| D | base_rendezvous_mgr.cc | 283 AllocationAttributes allocation_attr; in SameWorkerRecvDone()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | tf_allocator_adapter.cc | 42 tensorflow::AllocationAttributes attrs; in Allocate()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | bfc_allocator.h | 60 return AllocateRaw(alignment, num_bytes, AllocationAttributes()); in AllocateRaw()
64 const AllocationAttributes& allocation_attr) override;
97 const AllocationAttributes& allocation_attr);
|
| D | scoped_allocator.h | 105 const AllocationAttributes& allocator_attr) override { in AllocateRaw()
|
| D | base_collective_executor.cc | 144 AllocationAttributes empty; in TempChunk()
165 Tensor Scalar(Allocator* a, const AllocationAttributes& attr) const override { in Scalar()
|
| D | base_collective_executor.h | 62 const AllocationAttributes& attr) const = 0;
|
| D | rendezvous_mgr.cc | 85 AllocationAttributes aa; in SameWorkerRecvDone()
|
| D | ring_reducer.cc | 132 AllocationAttributes aa; in ContinueAfterInputCopy()
|
| D | bfc_allocator.cc | 202 const AllocationAttributes& allocation_attr) { in AllocateRawInternalWithRetry()
228 const AllocationAttributes& allocation_attr) { in AllocateRaw()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_event_mgr_test.cc | 355 {tensor_size}, AllocationAttributes())); in ReInit()
360 {tensor_size}, AllocationAttributes())); in ReInit()
366 {tensor_size}, AllocationAttributes())); in ReInit()
375 {tensor_size}, AllocationAttributes())); in ReInit()
|
| D | gpu_device.cc | 591 AllocationAttributes allocation_attr; in MaybeCopyTensorToGPU()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | buffer_map.cc | 117 tensorflow::AllocationAttributes()) in StringTfLiteTensorBuffer()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_gpu.h | 70 AllocationAttributes allocation_attr; in AllocateBytes()
|
| D | fft_ops.cc | 374 AllocationAttributes allocation_attr; in AllocateBytes()
|