Home
last modified time | relevance | path

Searched refs:map_computation (Results 1 – 13 of 13) 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() argument
58 kScalarShape, {last_value}, map_computation)); in MakeMappingComputation()
143 HloComputation* map_computation = in TEST_F() local
146 MakeMappingComputation(map_computation, /*callsites=*/5)); in TEST_F()
161 const CallGraphNode& map_node = call_graph->GetNode(map_computation); in TEST_F()
162 EXPECT_EQ(map_computation, map_node.computation()); in TEST_F()
168 EXPECT_EQ(5, call_graph->GetComputationCallers(map_computation).size()); in TEST_F()
H A Dhlo_computation_test.cc61 HloComputation* map_computation) { in CreateMapComputation() argument
66 HloInstruction::CreateMap(r0f32_, {param}, map_computation)); in CreateMapComputation()
85 auto map_computation = in TEST_F() local
88 EXPECT_THAT(map_computation->MakeEmbeddedComputationsList(), in TEST_F()
H A Dflatten_call_graph_test.cc48 HloComputation* map_computation, int64_t callsites) { in MakeMappingComputation() argument
55 kScalarShape, {last_value}, map_computation)); in MakeMappingComputation()
H A Dhlo_creation_utils.cc353 HloComputation* map_computation, in MakeMapHlo() argument
369 operand_shapes, map_computation->ComputeProgramShape(), map_dims)); in MakeMapHlo()
371 HloInstruction::CreateMap(map_shape, operands, map_computation), in MakeMapHlo()
H A Dbuffer_assignment_test.cc936 auto map_computation = in TEST_F() local
938 auto inner_last = map_computation->root_instruction(); in TEST_F()
945 HloInstruction::CreateMap(f32a100x10_, {param0}, map_computation)); in TEST_F()
1399 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F() local
1417 HloInstruction::CreateMap(vec_shape, {call}, map_computation)); in TEST_F()
1469 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F() local
1476 scalar_shape, {param}, map_computation, "call_name")); in TEST_F()
H A Dhlo_creation_utils.h170 HloComputation* map_computation,
H A Dhlo_instruction_test.cc1186 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F() local
1190 HloInstruction::CreateMap(r2f32, {x}, map_computation)); in TEST_F()
H A Dhlo_instruction.h612 HloComputation* map_computation);
H A Dhlo_instructions.cc1331 HloComputation* map_computation) in HloMapInstruction() argument
1336 AppendComputation(map_computation); in HloMapInstruction()
H A Dhlo_instructions.h1016 HloComputation* map_computation);
H A Dhlo_instruction.cc1166 HloComputation* map_computation) { in CreateMap() argument
1167 return std::make_unique<HloMapInstruction>(shape, operands, map_computation); in CreateMap()
H A Dalgebraic_simplifier.cc6703 auto* map_computation = map->to_apply(); in HandleMap() local
6704 auto* map_root = map_computation->root_instruction(); in HandleMap()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
H A Dimport.hlotxt657 %map_computation {
673 ROOT map = f32[4]{0} map(param0, param1), dimensions={0}, to_apply=%map_computation