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()
62 HloComputation* map_computation) { in CreateMapComputation()86 auto map_computation = in TEST_F() local
230 HloComputation* map_computation) { in MakeMapHlo()
745 auto map_computation = in TEST_F() local1208 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F() local
1026 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F() local
3692 auto* map_computation = map->to_apply(); in HandleMap() local
776 HloComputation* map_computation) { in CreateMap()
909 HloComputation* map_computation) in HloMapInstruction()