| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | call_graph_test.cc | 51 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 D | hlo_computation_test.cc | 61 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 D | flatten_call_graph_test.cc | 48 HloComputation* map_computation, int64_t callsites) { in MakeMappingComputation() argument
55 kScalarShape, {last_value}, map_computation)); in MakeMappingComputation()
|
| H A D | hlo_creation_utils.cc | 353 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 D | buffer_assignment_test.cc | 936 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 D | hlo_creation_utils.h | 170 HloComputation* map_computation,
|
| H A D | hlo_instruction_test.cc | 1186 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F() local
1190 HloInstruction::CreateMap(r2f32, {x}, map_computation)); in TEST_F()
|
| H A D | hlo_instruction.h | 612 HloComputation* map_computation);
|
| H A D | hlo_instructions.cc | 1331 HloComputation* map_computation) in HloMapInstruction() argument
1336 AppendComputation(map_computation); in HloMapInstruction()
|
| H A D | hlo_instructions.h | 1016 HloComputation* map_computation);
|
| H A D | hlo_instruction.cc | 1166 HloComputation* map_computation) { in CreateMap() argument
1167 return std::make_unique<HloMapInstruction>(shape, operands, map_computation); in CreateMap()
|
| H A D | algebraic_simplifier.cc | 6703 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 D | import.hlotxt | 657 %map_computation {
673 ROOT map = f32[4]{0} map(param0, param1), dimensions={0}, to_apply=%map_computation
|