Searched refs:retry_on_failure (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | device_memory_allocator.h | 178 bool retry_on_failure,
194 bool retry_on_failure) { in Allocate() argument
195 return Allocate(device_ordinal, size, retry_on_failure, in Allocate()
202 int device_ordinal, uint64 size, bool retry_on_failure = true,
204 return Allocate(device_ordinal, size, retry_on_failure, memory_space);
245 bool retry_on_failure,
|
| D | tf_allocator_adapter.cc | 39 int device_ordinal, uint64 size, bool retry_on_failure, in Allocate() argument
43 attrs.retry_on_failure = retry_on_failure; in Allocate()
|
| D | tf_allocator_adapter.h | 43 bool retry_on_failure,
82 bool retry_on_failure, in Allocate() argument
86 device_ordinal, size, retry_on_failure, memory_space); in Allocate()
|
| D | stream_executor_pimpl.cc | 898 int device_ordinal, uint64 size, bool retry_on_failure, in Allocate() argument
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | allocator.h | 42 AllocationAttributes(bool retry_on_failure, bool allocation_will_be_logged, in AllocationAttributes()
44 : retry_on_failure(retry_on_failure), in AllocationAttributes()
54 bool retry_on_failure = true; member
|
| D | op_kernel.cc | 713 /*retry_on_failure=*/allocation_attr.retry_on_failure, in allocate_tensor()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_test_base.cc | 46 bool retry_on_failure, in Allocate() argument
55 device_ordinal, size, retry_on_failure, memory_space); in Allocate()
|
| D | local_client_test_base.h | 54 bool retry_on_failure,
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_allocator.cc | 77 attributes.retry_on_failure = false; in allocate()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | c_api_decl.h | 104 bool retry_on_failure, int64_t memory_space,
|
| D | c_api_conversions.cc | 91 bool retry_on_failure, int64_t memory_space, in ToC()
96 ->Allocate(device_ordinal, size, retry_on_failure, memory_space); in ToC()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_gpu.h | 71 allocation_attr.retry_on_failure = false; in AllocateBytes()
|
| D | fft_ops.cc | 375 allocation_attr.retry_on_failure = false; in AllocateBytes()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | bfc_allocator.cc | 256 if (!allocation_attr.retry_on_failure) { in AllocateRaw()
|