Searched refs:subcomp_builder (Results 1 – 1 of 1) sorted by relevance
328 HloComputation::Builder subcomp_builder("reduction_subcomp"); in TEST_F() local331 subcomp_builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()334 subcomp_builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()336 subcomp_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()339 auto reduce_subcomp = module->AddEmbeddedComputation(subcomp_builder.Build()); in TEST_F()369 HloComputation::Builder subcomp_builder("reduction_subcomp"); in TEST_F() local372 subcomp_builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()375 subcomp_builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F()377 subcomp_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()380 auto reduce_subcomp = module->AddEmbeddedComputation(subcomp_builder.Build()); in TEST_F()