Searched refs:AllocationAttributes (Results 1 – 24 of 24) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | allocator.h | 39 struct AllocationAttributes { struct
40 AllocationAttributes() = default;
42 AllocationAttributes(bool retry_on_failure, bool allocation_will_be_logged, in AllocationAttributes() argument
66 TF_DISALLOW_COPY_AND_ASSIGN(AllocationAttributes); argument
202 const AllocationAttributes& allocation_attr) { in AllocateRaw()
311 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 | 1013 const AllocationAttributes& allocation_attr);
1017 AllocationAttributes()); in allocate_temp()
1280 AllocationAttributes()); in allocate_tensor()
1285 const AllocationAttributes& allocation_attr);
|
| D | op_kernel.cc | 263 AllocationAttributes attr; in allocate_temp()
288 AllocationAttributes attr; in allocate_temp()
708 AllocatorAttributes attr, const AllocationAttributes& allocation_attr) { in allocate_tensor()
712 AllocationAttributes( in allocate_tensor()
778 const AllocationAttributes& allocation_attr) { in allocate_temp()
|
| D | tensor.h | 153 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 | 268 AllocationAttributes allocation_attr; in SameWorkerRecvDone()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | tf_allocator_adapter.cc | 42 tensorflow::AllocationAttributes attrs; in Allocate()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_allocator.cc | 76 AllocationAttributes attributes; 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.h | 63 const AllocationAttributes& attr) const = 0;
|
| D | base_collective_executor.cc | 153 AllocationAttributes empty; in TempChunk()
175 Tensor Scalar(Allocator* a, const AllocationAttributes& attr) const override { in Scalar()
|
| D | rendezvous_mgr.cc | 86 AllocationAttributes aa; in SameWorkerRecvDone()
|
| D | ring_reducer.cc | 131 AllocationAttributes aa; in ContinueAfterInputCopy()
|
| D | bfc_allocator.cc | 228 const AllocationAttributes& allocation_attr) { in AllocateRawInternalWithRetry()
254 const AllocationAttributes& allocation_attr) { in AllocateRaw()
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_event_mgr_test.cc | 210 {tensor_size}, AllocationAttributes())); in ReInit()
215 {tensor_size}, AllocationAttributes())); in ReInit()
221 {tensor_size}, AllocationAttributes())); in ReInit()
230 {tensor_size}, AllocationAttributes())); in ReInit()
|
| /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()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 648 AllocationAttributes allocation_attr; in MaybeCopyTensorToGPU()
|