Home
last modified time | relevance | path

Searched refs:GetCompilationDevice (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Ddevice_util.h125 DeviceRegistration* GetCompilationDevice(DeviceId device) const { in GetCompilationDevice() function
129 xla::StatusOr<DeviceRegistration*> GetCompilationDevice( in GetCompilationDevice() function
132 return GetCompilationDevice(device_id); in GetCompilationDevice()
Dxla_platform_info.cc68 if (!XlaOpRegistry::GetCompilationDevice(platform_info.device_type().type(), in BuildXlaCompilationCache()
Dxla_kernel_creator.cc44 CHECK(XlaOpRegistry::GetCompilationDevice(device->device_type(), in IsCompilable()
Ddevice_util.cc71 if (!XlaOpRegistry::GetCompilationDevice(device_type->type(), in GetIdFor()
Dpartially_decluster_pass.cc241 if (XlaOpRegistry::GetCompilationDevice(device_type.type(), &registration)) { in IsMustCompileDevice()
Dmark_for_compilation_pass.cc596 device_info_cache->GetCompilationDevice(n.assigned_device_name())); in IgnoreResourceOpForSafetyAnalysis()
1192 if (!XlaOpRegistry::GetCompilationDevice(device_type.type(), in FindCompilationCandidates()
1634 device_info_cache_.GetCompilationDevice(chosen_device); in ShouldCompileClusterImpl()
Dbuild_xla_ops_pass.cc289 device_info_cache.GetCompilationDevice(device); in DeviceRequiresCompilation()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_registry.h166 static bool GetCompilationDevice(const string& device_name,
Dxla_op_registry.cc144 /* static */ bool XlaOpRegistry::GetCompilationDevice( in GetCompilationDevice() function in tensorflow::XlaOpRegistry