Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Ddevice_memory_allocator.h70 virtual bool AllowsAsynchronousDeallocation() const = 0;
93 bool AllowsAsynchronousDeallocation() const override;
Ddevice_memory_allocator.cc80 bool StreamExecutorMemoryAllocator::AllowsAsynchronousDeallocation() const { in AllowsAsynchronousDeallocation() function in xla::StreamExecutorMemoryAllocator
Dshaped_buffer_test.cc88 bool AllowsAsynchronousDeallocation() const override { return false; } in AllowsAsynchronousDeallocation() function in xla::__anonfef8839d0111::TestAllocator
/external/tensorflow/tensorflow/compiler/jit/
Dxla_launch_util.h126 bool AllowsAsynchronousDeallocation() const override { return true; } in AllowsAsynchronousDeallocation() function
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc283 !memory_allocator->AllowsAsynchronousDeallocation(); in ExecuteOnStream()