Searched refs:GetCompilationDevice (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | mark_for_compilation_pass.cc | 503 XlaOpRegistry::GetCompilationDevice(device_type.type(), ®istration)); in FindCompilationCandidates()
656 CHECK(XlaOpRegistry::GetCompilationDevice(device->device_type(), in IsCompilable()
716 if (!XlaOpRegistry::GetCompilationDevice(device_type.type(), in Run()
936 if (!XlaOpRegistry::GetCompilationDevice(device_type.type(), ®istration)) { in IgnoreResourceOpForSafetyAnalysis()
1005 XlaOpRegistry::GetCompilationDevice(device_type.type(), ®istration)) in AreDevicesCompatible()
1036 XlaOpRegistry::GetCompilationDevice(device_type.type(), ®istration)) in ShouldCompileClusterImpl()
|
| D | partially_decluster_pass.cc | 229 if (XlaOpRegistry::GetCompilationDevice(device_type.type(), ®istration)) { in IsMustCompileDevice()
|
| D | build_xla_ops_pass.cc | 222 if (!XlaOpRegistry::GetCompilationDevice(device_type.type(), ®istration)) { in DeviceRequiresCompilation()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_registry.h | 127 static bool GetCompilationDevice(const string& device_name,
|
| D | xla_op_registry.cc | 131 /* static */ bool XlaOpRegistry::GetCompilationDevice( in GetCompilationDevice() function in tensorflow::XlaOpRegistry
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 244 if (!XlaOpRegistry::GetCompilationDevice(platform_info.device_type().type(), in BuildCompilationCache()
|