Searched refs:ExpectScalarEq (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_lib_test.cc | 202 ExpectScalarEq<float>(parallel_result->tensor(0), 3.); in TEST()
203 ExpectScalarEq<float>(parallel_result->tensor(1), 3.); in TEST()
311 ExpectScalarEq<float>(float_tensors->tensor(0), 10.0); in TEST()
312 ExpectScalarEq<float>(float_tensors->tensor(1), 11.0); in TEST()
320 ExpectScalarEq<int>(int_tensors->tensor(0), 5); in TEST()
321 ExpectScalarEq<int>(int_tensors->tensor(1), 6); in TEST()
|
| D | parallel_device_test.cc | 148 ExpectScalarEq<float>(components[0].get(), 3.); in TEST()
149 ExpectScalarEq<float>(components[1].get(), 3.); in TEST()
282 ExpectScalarEq<float>(second_components[1].get(), 9.); in TEST()
296 ExpectScalarEq<float>(first_components[0].get(), 3.); in TEST()
297 ExpectScalarEq<float>(first_components[1].get(), 6.); in TEST()
459 ExpectScalarEq<float>(result_components[0].get(), 3.); in TestCollective()
460 ExpectScalarEq<float>(result_components[1].get(), 3.); in TestCollective()
568 ExpectScalarEq<float>(result_components[0].get(), 7. * 9.); in TEST()
569 ExpectScalarEq<float>(result_components[1].get(), 7. * 9.); in TEST()
|
| D | parallel_device_testlib.cc | 244 ExpectScalarEq<float>(components[0].get(), 20.); in BasicTestsForTwoDevices()
245 ExpectScalarEq<float>(components[1].get(), 20.); in BasicTestsForTwoDevices()
274 ExpectScalarEq<float>(components[0].get(), 23.); in BasicTestsForTwoDevices()
275 ExpectScalarEq<float>(components[1].get(), 18.); in BasicTestsForTwoDevices()
|
| D | parallel_device_testlib.h | 101 void ExpectScalarEq(TFE_TensorHandle* handle, value_type expected_value);
140 void ExpectScalarEq(TFE_TensorHandle* handle, value_type expected_value) { in ExpectScalarEq() function
|
| D | parallel_device_remote_test.cc | 144 ExpectScalarEq<float>(out_components[0].get(), 9.); in TEST()
145 ExpectScalarEq<float>(out_components[1].get(), 4.); in TEST()
|