Home
last modified time | relevance | path

Searched defs:map_computation (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dcall_graph_test.cc51 HloComputation* map_computation, int64_t callsites) { in MakeMappingComputation()
143 HloComputation* map_computation = in TEST_F() local
H A Dhlo_computation_test.cc61 HloComputation* map_computation) { in CreateMapComputation()
85 auto map_computation = in TEST_F() local
H A Dflatten_call_graph_test.cc48 HloComputation* map_computation, int64_t callsites) { in MakeMappingComputation()
H A Dbuffer_assignment_test.cc936 auto map_computation = in TEST_F() local
1399 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F() local
1469 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F() local
H A Dhlo_creation_utils.cc353 HloComputation* map_computation, in MakeMapHlo()
H A Dhlo_instruction_test.cc1186 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F() local
H A Dhlo_instructions.cc1331 HloComputation* map_computation) in HloMapInstruction()
H A Dhlo_instruction.cc1166 HloComputation* map_computation) { in CreateMap()
H A Dalgebraic_simplifier.cc6703 auto* map_computation = map->to_apply(); in HandleMap() local