Searched refs:parallel_result (Results 1 – 3 of 3) sorted by relevance
335 parallel_result = computation()340 nest.assert_same_structure(parallel_result, non_parallel_result)342 parallel_flat = nest.flatten(parallel_result)360 parallel_result = f(y)361 self.assertAllClose([[2., 3.]] * 2, self.device.unpack(parallel_result))
201 ParallelTensor* parallel_result = (*outputs)[0].get(); in TEST() local202 ExpectScalarEq<float>(parallel_result->tensor(0), 3.); in TEST()203 ExpectScalarEq<float>(parallel_result->tensor(1), 3.); in TEST()
173 for (std::unique_ptr<ParallelTensor>& parallel_result : parallel_results) { in ExecuteWithSpecialOps()175 MaybeParallelTensorOwned(std::move(parallel_result))); in ExecuteWithSpecialOps()