Home
last modified time | relevance | path

Searched defs:xla_allocator (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.h42 std::unique_ptr<XlaAllocator> xla_allocator, in XlaPlatformInfo()
Dxla_ops.cc64 std::unique_ptr<XlaAllocator> xla_allocator; in PlatformInfoFromContext() local
/external/tensorflow/tensorflow/compiler/jit/
Dxla_launch_util.cc195 xla::LocalClient* client, xla::DeviceMemoryAllocator* xla_allocator, in XlaComputationLaunchContext()