Home
last modified time | relevance | path

Searched refs:device_allocator_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.h48 device_allocator_(device_allocator) { in XlaPlatformInfo()
49 CHECK((device_allocator_ != nullptr) ^ (xla_allocator_.get() != nullptr)); in XlaPlatformInfo()
59 return device_allocator_ ? device_allocator_ : xla_allocator_.get(); in allocator()
89 xla::DeviceMemoryAllocator* device_allocator_; variable
/external/tensorflow/tensorflow/compiler/xla/client/
Dexecutable_build_options.cc26 device_allocator_ = allocator; in set_device_allocator()
31 return device_allocator_; in device_allocator()
Dexecutable_build_options.h80 DeviceMemoryAllocator* device_allocator_ = nullptr; variable
/external/tensorflow/tensorflow/compiler/xla/service/
Dcompiler.h78 DeviceMemoryAllocator* device_allocator() const { return device_allocator_; } in device_allocator()
80 device_allocator_ = device_allocator; in set_device_allocator()
101 DeviceMemoryAllocator* device_allocator_ = nullptr;