Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/client/
Dexecutable_build_options.h62 ExecutableBuildOptions& set_device_allocator(
Dexecutable_build_options.cc24 ExecutableBuildOptions& ExecutableBuildOptions::set_device_allocator( in set_device_allocator() function in xla::ExecutableBuildOptions
/external/tensorflow/tensorflow/compiler/xla/service/
Dcompiler.h79 void set_device_allocator(DeviceMemoryAllocator* device_allocator) { in set_device_allocator() function
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_compile_ops.cc125 build_options.set_device_allocator(device_ref.backend()->memory_allocator()); in Compile()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_compilation_cache.cc156 build_options.set_device_allocator(options.device_allocator); in BuildExecutable()