Searched refs:parallel_result (Results 1 – 4 of 4) sorted by relevance
470 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))
202 ParallelTensor* parallel_result = (*outputs)[0].get(); in TEST() local203 ExpectScalarEq<float>(parallel_result->tensor(0), 3.); in TEST()204 ExpectScalarEq<float>(parallel_result->tensor(1), 3.); in TEST()
193 for (std::unique_ptr<ParallelTensor>& parallel_result : parallel_results) { in ExecuteWithSpecialOps()195 MaybeParallelTensorOwned(std::move(parallel_result))); in ExecuteWithSpecialOps()
173 parallel_result = test_target_modules(179 return parallel_result and serial_result