Searched refs:reduced_values (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops_test.py | 244 reduced_values = collective.reduce(options.reduce_op, per_replica_value,
247 reduced_values = self.as_list(reduced_values)
248 self.assertAllEqual(devices, [v.device for v in reduced_values])
249 return [ops.convert_to_tensor(v) for v in reduced_values]
290 reduced_values = collective.batch_reduce(options.reduce_op,
293 reduced_values = [self.as_list(v) for v in reduced_values]
294 for v in reduced_values:
296 return nest.map_structure(ops.convert_to_tensor, reduced_values)
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_lib_test.cc | 187 std::unique_ptr<ParallelTensor> reduced_values = in TEST() local
196 context.get(), {reduced_values.get(), run_collective.get()}, in TEST()
208 {reduced_values.get(), run_collective.get()}, in TEST()
|