Searched defs:map_computation (Results 1 – 9 of 9) sorted by relevance
51 HloComputation* map_computation, int64_t callsites) { in MakeMappingComputation()143 HloComputation* map_computation = in TEST_F() local
48 HloComputation* map_computation, int64_t callsites) { in MakeMappingComputation()
61 HloComputation* map_computation) { in CreateMapComputation()85 auto map_computation = in TEST_F() local
353 HloComputation* map_computation, in MakeMapHlo()
936 auto map_computation = in TEST_F() local1399 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F() local1469 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F() local
1186 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F() local
1166 HloComputation* map_computation) { in CreateMap()
1331 HloComputation* map_computation) in HloMapInstruction()
6703 auto* map_computation = map->to_apply(); in HandleMap() local