Home
last modified time | relevance | path

Searched refs:xla_device_metadata (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dxla_platform_info.cc25 if (platform_info.xla_device_metadata()) { in BuildXlaCompilationCache()
27 platform_info.xla_device_metadata()->client(), in BuildXlaCompilationCache()
28 platform_info.xla_device_metadata()->jit_device_type()); in BuildXlaCompilationCache()
81 const XlaDevice::Metadata* xla_device_metadata = nullptr; in XlaPlatformInfoFromDevice() local
91 } else if (XlaDevice::GetMetadataFromDevice(device, &xla_device_metadata) in XlaPlatformInfoFromDevice()
102 platform_id = xla_device_metadata->platform()->id(); in XlaPlatformInfoFromDevice()
104 xla_device_metadata->client()->backend().memory_allocator(); in XlaPlatformInfoFromDevice()
108 xla_device_metadata, custom_allocator); in XlaPlatformInfoFromDevice()
147 if (platform_info.xla_device_metadata()) { in GenerateCompilerOptions()
149 platform_info.xla_device_metadata()->shape_representation_fn(); in GenerateCompilerOptions()
Dxla_platform_info.h34 const XlaDevice::Metadata* xla_device_metadata, in XlaPlatformInfo() argument
38 xla_device_metadata_(xla_device_metadata), in XlaPlatformInfo()
60 const XlaDevice::Metadata* xla_device_metadata() const { in xla_device_metadata() function
63 bool is_on_xla_device() const { return xla_device_metadata() != nullptr; } in is_on_xla_device()
Dxla_compile_on_demand_op.cc58 /*allocate_xla_tensors=*/platform_info_.xla_device_metadata() != nullptr, in Run()
59 platform_info_.xla_device_metadata() in Run()
60 ? platform_info_.xla_device_metadata()->UseMultipleStreams() in Run()