Searched refs:inner_computation (Results 1 – 3 of 3) sorted by relevance
55 HloComputation* inner_computation = in TEST_F() local62 HloInstruction::CreateCall(r0f32, {}, inner_computation)); in TEST_F()
306 HloComputation* inner_computation = module->AddEmbeddedComputation( in TEST_F() local310 while_cond, /*while_body=*/inner_computation, in TEST_F()318 EXPECT_EQ(inner_computation->instruction_count(), 8); in TEST_F()330 EXPECT_EQ(inner_computation->instruction_count(), 9); in TEST_F()
420 HloComputation* inner_computation = in TEST_P() local430 scalar_shape_, {outer_param1, outer_param0}, inner_computation)); in TEST_P()