Searched defs:map_computation (Results 1 – 9 of 9) sorted by relevance
50 HloComputation* map_computation, int64 callsites) { in MakeMappingComputation()142 HloComputation* map_computation = in TEST_F() local
48 HloComputation* map_computation, int64 callsites) { in MakeMappingComputation()
61 HloComputation* map_computation) { in CreateMapComputation()85 auto map_computation = in TEST_F() local
277 HloComputation* map_computation) { in MakeMapHlo()
924 auto map_computation = in TEST_F() local1387 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F() local
1029 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F() local
4394 auto* map_computation = map->to_apply(); in HandleMap() local
854 HloComputation* map_computation) { in CreateMap()
1003 HloComputation* map_computation) in HloMapInstruction()