Home
last modified time | relevance | path

Searched refs:no_retry_on_failure (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dallocator.h40 AllocationAttributes(bool no_retry_on_failure, bool allocation_will_be_logged, in AllocationAttributes()
42 : no_retry_on_failure(no_retry_on_failure), in AllocationAttributes()
50 bool no_retry_on_failure = false; member
Dop_kernel.cc754 AllocationAttributes(allocation_attr.no_retry_on_failure, in allocate_tensor()
/external/tensorflow/tensorflow/stream_executor/
Dtf_allocator_adapter.cc43 attrs.no_retry_on_failure = !retry_on_failure; in Allocate()
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_gpu.h71 allocation_attr.no_retry_on_failure = true; in AllocateBytes()
Dfft_ops.cc375 allocation_attr.no_retry_on_failure = true; in AllocateBytes()
/external/tensorflow/tensorflow/core/common_runtime/
Dbfc_allocator.cc230 if (allocation_attr.no_retry_on_failure) { in AllocateRaw()