Searched refs:SupportedDeviceTypesForNode (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel_test.cc | 210 TF_ASSERT_OK(SupportedDeviceTypesForNode(DeviceTypes(), ndef, &devs)); in TEST_F()
221 TF_ASSERT_OK(SupportedDeviceTypesForNode(DeviceTypes(), ndef, &devs)); in TEST_F()
230 TF_ASSERT_OK(SupportedDeviceTypesForNode(DeviceTypes(), ndef, &devs)); in TEST_F()
238 TF_ASSERT_OK(SupportedDeviceTypesForNode(DeviceTypes(), ndef, &devs)); in TEST_F()
246 TF_ASSERT_OK(SupportedDeviceTypesForNode(DeviceTypes(), ndef, &devs)); in TEST_F()
256 TF_ASSERT_OK(SupportedDeviceTypesForNode(DeviceTypes(), ndef, &devs)); in TEST_F()
449 TF_EXPECT_OK(SupportedDeviceTypesForNode(DeviceTypes(), def, &devices));
483 TF_EXPECT_OK(SupportedDeviceTypesForNode(DeviceTypes(), def, &devices)); in ExpectFailure()
489 SupportedDeviceTypesForNode(DeviceTypes(), def, &devices); in ExpectFailure()
599 Status status = SupportedDeviceTypesForNode(DeviceTypes(), ndef, &devs); in TEST_F()
[all …]
|
| D | op_kernel.h | 1329 Status SupportedDeviceTypesForNode(
|
| D | op_kernel.cc | 1172 Status SupportedDeviceTypesForNode( in SupportedDeviceTypesForNode() function
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | colocation_graph.cc | 171 return SupportedDeviceTypesForNode(types, node.def(), in SetParentAndSupportedDevices()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute.cc | 207 TF_RETURN_IF_ERROR(SupportedDeviceTypesForNode( in SelectDevice()
|