Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Ddevice_memory_allocator.h51 bool retry_on_failure) = 0;
86 bool retry_on_failure) override;
Ddevice_memory_allocator.cc34 int device_ordinal, uint64 size, bool retry_on_failure) { in Allocate() argument
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_test_base.cc41 bool retry_on_failure) { in Allocate() argument
49 retry_on_failure); in Allocate()
Dlocal_client_test_base.h50 bool retry_on_failure) override;
/external/tensorflow/tensorflow/compiler/jit/
Dxla_launch_util.h116 int device_ordinal, uint64 size, bool retry_on_failure) override;
Dxla_launch_util.cc174 int device_ordinal, uint64 size, bool retry_on_failure) { in Allocate() argument
176 attrs.no_retry_on_failure = !retry_on_failure; in Allocate()