Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_gpu.h58 allocation_attr.no_retry_on_failure = true; in AllocateBytes()
Dfft_ops.cc326 allocation_attr.no_retry_on_failure = true; in AllocateBytes()
/external/tensorflow/tensorflow/core/framework/
Dallocator.h44 bool no_retry_on_failure = false; member
/external/tensorflow/tensorflow/compiler/jit/
Dxla_launch_util.cc176 attrs.no_retry_on_failure = !retry_on_failure; in Allocate()
/external/tensorflow/tensorflow/core/common_runtime/
Dbfc_allocator.cc215 if (allocation_attr.no_retry_on_failure) { in AllocateRaw()