Searched refs:ExtractPerDeviceValues (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_testlib.cc | 159 void ExtractPerDeviceValues( in ExtractPerDeviceValues() function
241 ExtractPerDeviceValues(context, read.get(), &components, status.get()); in BasicTestsForTwoDevices()
271 ExtractPerDeviceValues(context, read.get(), &components, status.get()); in BasicTestsForTwoDevices()
|
| D | parallel_device_test.cc | 143 ExtractPerDeviceValues(context.get(), device_value.get(), &components, in TEST()
278 ExtractPerDeviceValues(context.get(), multiply_result.get(), in TEST()
294 ExtractPerDeviceValues(context.get(), second_components[0].get(), in TEST()
343 ExtractPerDeviceValues(context.get(), combined_value.get(), in TEST()
456 ExtractPerDeviceValues(context.get(), reduced.get(), &result_components, in TestCollective()
565 ExtractPerDeviceValues(context.get(), reduced.get(), &result_components, in TEST()
|
| D | parallel_device_testlib.h | 85 void ExtractPerDeviceValues(
|
| D | parallel_device_remote_test.cc | 140 ExtractPerDeviceValues(context.get(), multiply_result.get(), in TEST()
|