Searched defs:map_computation (Results 1 – 9 of 9) sorted by relevance
50 HloComputation* map_computation, int64_t callsites) { in MakeMappingComputation()142 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
302 HloComputation* map_computation) { in MakeMapHlo()
919 auto map_computation = in TEST_F() local1382 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F() local1452 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F() local
5065 auto map_computation = computation_->parent()->AddEmbeddedComputation( in HandleReduceWindow() local5884 auto* map_computation = map->to_apply(); in HandleMap() local
1087 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F() local
1096 HloComputation* map_computation) { in CreateMap()
1242 HloComputation* map_computation) in HloMapInstruction()