Searched refs:SupportedDeviceTypesForNode (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel_test.cc | 218 TF_ASSERT_OK(SupportedDeviceTypesForNode(DeviceTypes(), ndef, &devs)); in TEST_F()
234 TF_ASSERT_OK(SupportedDeviceTypesForNode(DeviceTypes(), ndef, &devs, in TEST_F()
249 TF_ASSERT_OK(SupportedDeviceTypesForNode(DeviceTypes(), ndef, &devs, in TEST_F()
261 TF_ASSERT_OK(SupportedDeviceTypesForNode(DeviceTypes(), ndef, &devs)); in TEST_F()
270 TF_ASSERT_OK(SupportedDeviceTypesForNode(DeviceTypes(), ndef, &devs)); in TEST_F()
278 TF_ASSERT_OK(SupportedDeviceTypesForNode(DeviceTypes(), ndef, &devs)); in TEST_F()
286 TF_ASSERT_OK(SupportedDeviceTypesForNode(DeviceTypes(), ndef, &devs)); in TEST_F()
296 TF_ASSERT_OK(SupportedDeviceTypesForNode(DeviceTypes(), ndef, &devs)); in TEST_F()
524 TF_EXPECT_OK(SupportedDeviceTypesForNode(DeviceTypes(), def, &devices));
558 TF_EXPECT_OK(SupportedDeviceTypesForNode(DeviceTypes(), def, &devices)); in ExpectFailure()
[all …]
|
| D | op_kernel.h | 1402 Status SupportedDeviceTypesForNode(
|
| D | op_kernel.cc | 1453 Status SupportedDeviceTypesForNode( in SupportedDeviceTypesForNode() function
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | colocation_graph.cc | 154 return SupportedDeviceTypesForNode( in SetParentAndSupportedDevices()
1352 SupportedDeviceTypesForNode(device_types_, node->def(), &supported_types, in DebugInfo()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 274 TF_RETURN_IF_ERROR(SupportedDeviceTypesForNode( in SelectDevice()
|