Searched refs:map_computation (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | call_graph_test.cc | 50 HloComputation* map_computation, int64 callsites) { in MakeMappingComputation() argument
57 kScalarShape, {last_value}, map_computation)); in MakeMappingComputation()
142 HloComputation* map_computation = in TEST_F() local
145 MakeMappingComputation(map_computation, /*callsites=*/5)); in TEST_F()
159 const CallGraphNode& map_node = call_graph->GetNode(map_computation); in TEST_F()
160 EXPECT_EQ(map_computation, map_node.computation()); in TEST_F()
|
| D | flatten_call_graph_test.cc | 48 HloComputation* map_computation, int64 callsites) { in MakeMappingComputation() argument
55 kScalarShape, {last_value}, map_computation)); in MakeMappingComputation()
|
| D | hlo_creation_utils.cc | 230 HloComputation* map_computation) { in MakeMapHlo() argument
245 operand_shapes, map_computation->ComputeProgramShape(), map_dims)); in MakeMapHlo()
247 HloInstruction::CreateMap(map_shape, operands, map_computation)); in MakeMapHlo()
|
| D | hlo_computation_test.cc | 62 HloComputation* map_computation) { in CreateMapComputation() argument
67 HloInstruction::CreateMap(r0f32_, {param}, map_computation)); in CreateMapComputation()
86 auto map_computation = in TEST_F() local
89 EXPECT_THAT(map_computation->MakeEmbeddedComputationsList(), in TEST_F()
|
| D | hlo_creation_utils.h | 115 HloComputation* map_computation);
|
| D | buffer_assignment_test.cc | 745 auto map_computation = in TEST_F() local
747 auto inner_last = map_computation->root_instruction(); in TEST_F()
754 HloInstruction::CreateMap(f32a100x10_, {param0}, map_computation)); in TEST_F()
1208 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F() local
1226 HloInstruction::CreateMap(vec_shape, {call}, map_computation)); in TEST_F()
|
| D | hlo_instruction_test.cc | 1026 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F() local
1030 HloInstruction::CreateMap(r2f32, {x}, map_computation)); in TEST_F()
|
| D | hlo_instruction.h | 433 HloComputation* map_computation);
|
| D | hlo_instructions.cc | 909 HloComputation* map_computation) in HloMapInstruction() argument
914 AppendComputation(map_computation); in HloMapInstruction()
|
| D | hlo_instructions.h | 584 HloComputation* map_computation);
|
| D | hlo_instruction.cc | 776 HloComputation* map_computation) { in CreateMap() argument
777 return absl::make_unique<HloMapInstruction>(shape, operands, map_computation); in CreateMap()
|
| D | algebraic_simplifier.cc | 3692 auto* map_computation = map->to_apply(); in HandleMap() local
3693 auto* map_root = map_computation->root_instruction(); in HandleMap()
|