/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
|