Searched refs:DeviceToDeviceType (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | partially_decluster_pass.cc | 60 DeviceToDeviceType(n->assigned_device_name(), &device_type)); in FindNodesToDecluster()
81 DeviceToDeviceType(dst->assigned_device_name(), &dst_device_type)); in FindNodesToDecluster()
240 DeviceToDeviceType(n->assigned_device_name(), &device_type)); in MustCompileNode()
|
| D | xla_cluster_util.h | 50 Status DeviceToDeviceType(const string& device, DeviceType* device_type);
|
| D | mark_for_compilation_pass.cc | 492 DeviceToDeviceType(node->assigned_device_name(), &device_type)); in FindCompilationCandidates()
933 DeviceToDeviceType(n.assigned_device_name(), &device_type)); in IgnoreResourceOpForSafetyAnalysis()
1003 TF_RETURN_IF_ERROR(DeviceToDeviceType(chosen_device, &device_type)); in AreDevicesCompatible()
1034 TF_RETURN_IF_ERROR(DeviceToDeviceType(chosen_device, &device_type)); in ShouldCompileClusterImpl()
|
| D | xla_fusion_optimizer.cc | 178 TF_RETURN_IF_ERROR(DeviceToDeviceType(node->def().device(), &device_type)); in Optimize()
|
| D | xla_cluster_util.cc | 87 Status DeviceToDeviceType(const string& device, DeviceType* device_type) { in DeviceToDeviceType() function
|
| D | build_xla_ops_pass.cc | 220 TF_RETURN_IF_ERROR(DeviceToDeviceType(device, &device_type)); in DeviceRequiresCompilation()
|