Searched refs:xla_device_metadata (Results 1 – 2 of 2) sorted by relevance
41 const XlaDevice::Metadata* xla_device_metadata, in XlaPlatformInfo() argument46 xla_device_metadata_(xla_device_metadata), in XlaPlatformInfo()69 const XlaDevice::Metadata* xla_device_metadata() const { in xla_device_metadata() function72 bool is_on_xla_device() const { return xla_device_metadata() != nullptr; } in is_on_xla_device()
63 const XlaDevice::Metadata* xla_device_metadata = nullptr; in PlatformInfoFromContext() local75 } else if (XlaDevice::GetMetadata(ctx, &xla_device_metadata).ok()) { in PlatformInfoFromContext()86 platform_id = xla_device_metadata->platform()->id(); in PlatformInfoFromContext()88 xla_device_metadata->client()->backend().memory_allocator(); in PlatformInfoFromContext()100 return XlaPlatformInfo(device_type, platform_id, xla_device_metadata, in PlatformInfoFromContext()201 if (platform_info.xla_device_metadata()) { in BuildCompilationCache()203 platform_info.xla_device_metadata()->client(), in BuildCompilationCache()204 platform_info.xla_device_metadata()->jit_device_type()); in BuildCompilationCache()294 if (platform_info.xla_device_metadata()) { in CompileToLocalExecutable()296 platform_info.xla_device_metadata()->shape_representation_fn(); in CompileToLocalExecutable()