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.h60 allocation_attr.no_retry_on_failure = true; in AllocateBytes()
Dfft_ops.cc317 allocation_attr.no_retry_on_failure = true; in AllocateBytes()
/external/tensorflow/tensorflow/core/framework/
Dallocator.h39 bool no_retry_on_failure = false; member
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_launch_op.cc94 allocation_attrs.no_retry_on_failure = !retry_on_failure; in Allocate()
/external/tensorflow/tensorflow/core/common_runtime/
Dbfc_allocator.cc208 if (allocation_attr.no_retry_on_failure) { in AllocateRaw()