Searched refs:call_computation (Results 1 – 2 of 2) sorted by relevance
2738 HloComputation* call_computation = in TEST_P() local2753 HloInstruction::CreateCall(shape, {add1, negate8}, call_computation)); in TEST_P()2765 call_computation, in TEST_P()2813 HloComputation* call_computation = in TEST_P() local2824 HloInstruction::CreateCall(shape, {add1}, call_computation)); in TEST_P()2832 call_computation, in TEST_P()
1388 auto call_computation = module->AddEmbeddedComputation(call_builder.Build()); in TEST_F() local1396 HloInstruction::CreateCall(vec_shape, {param}, call_computation)); in TEST_F()