Searched refs:map_computation (Results 1 – 8 of 8) 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()
138 HloComputation* map_computation = in TEST_F() local
141 MakeMappingComputation(map_computation, /*callsites=*/5)); in TEST_F()
154 const CallGraphNode& map_node = call_graph->GetNode(map_computation); in TEST_F()
155 EXPECT_EQ(map_computation, map_node.computation()); in TEST_F()
|
| D | hlo_tfgraph_builder_test.cc | 45 HloComputation *map_computation) { in CreateMapComputation() argument
50 HloInstruction::CreateMap(r0f32_, {param}, map_computation)); in CreateMapComputation()
|
| D | hlo_computation_test.cc | 55 HloComputation* map_computation) { in CreateMapComputation() argument
60 HloInstruction::CreateMap(r0f32_, {param}, map_computation)); in CreateMapComputation()
79 auto map_computation = in TEST_F() local
82 EXPECT_THAT(map_computation->MakeEmbeddedComputationsList(), 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 | buffer_assignment_test.cc | 582 auto map_computation = in TEST_F() local
584 auto inner_last = map_computation->root_instruction(); in TEST_F()
591 HloInstruction::CreateMap(f32a100x10_, {param0}, map_computation)); in TEST_F()
1045 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F() local
1063 HloInstruction::CreateMap(vec_shape, {call}, map_computation)); in TEST_F()
|
| D | hlo_instruction.h | 252 HloComputation* map_computation,
|
| D | hlo_instruction.cc | 310 HloComputation* map_computation, in CreateMap() argument
317 instruction->called_computations_.push_back(map_computation); in CreateMap()
|
| D | user_computation.cc | 3033 HloComputation* map_computation = in Visit() local
3036 request.output_shape(), operands, map_computation)); in Visit()
|