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
302 HloComputation* map_computation) { in MakeMapHlo()
917 auto map_computation = in TEST_F() local1380 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F() local
1087 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F() local
979 HloComputation* map_computation) { in CreateMap()
5571 auto* map_computation = map->to_apply(); in HandleMap() local
1142 HloComputation* map_computation) in HloMapInstruction()