Home
last modified time | relevance | path

Searched refs:parallel_result (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/parallel_device/
H A Dparallel_device_test.py470 parallel_result = computation()
475 nest.assert_same_structure(parallel_result, non_parallel_result)
477 parallel_flat = nest.flatten(parallel_result)
495 parallel_result = f(y)
496 self.assertAllClose([[2., 3.]] * 2, self.device.unpack(parallel_result))
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/parallel_device/
H A Dparallel_device_lib_test.cc202 ParallelTensor* parallel_result = (*outputs)[0].get(); in TEST() local
203 ExpectScalarEq<float>(parallel_result->tensor(0), 3.); in TEST()
204 ExpectScalarEq<float>(parallel_result->tensor(1), 3.); in TEST()
H A Dparallel_device.cc193 for (std::unique_ptr<ParallelTensor>& parallel_result : parallel_results) { in ExecuteWithSpecialOps()
195 MaybeParallelTensorOwned(std::move(parallel_result))); in ExecuteWithSpecialOps()
/aosp_15_r20/external/crosvm/tools/windows/
H A Dbuild_test.py173 parallel_result = test_target_modules(
179 return parallel_result and serial_result