Searched refs:call_computation (Results 1 – 2 of 2) sorted by relevance
1216 auto call_computation = module->AddEmbeddedComputation(call_builder.Build()); in TEST_F() local1224 HloInstruction::CreateCall(vec_shape, {param}, call_computation)); in TEST_F()
1357 def call_computation(computation, function2441 tensors = call_computation(