Searched refs:CreateMap (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | map_inliner_test.cc | 59 HloInstruction::CreateMap(lhs->shape(), {lhs, rhs}, max_f32.get())); in TEST_F()
93 HloInstruction::CreateMap(lhs->shape(), {lhs}, const2_f32.get())); in TEST_F()
131 HloInstruction::CreateMap(lhs->shape(), {lhs, rhs}, max_f32.get())); in TEST_F()
163 HloInstruction::CreateMap(lhs->shape(), {lhs, rhs}, param_f32.get())); in TEST_F()
|
| D | hlo_computation_test.cc | 67 HloInstruction::CreateMap(r0f32_, {param}, map_computation)); in CreateMapComputation()
107 HloInstruction::CreateMap(r0f32_, {param}, map1_computation)); in TEST_F()
109 HloInstruction::CreateMap(r0f32_, {param}, map2_computation)); in TEST_F()
|
| D | flatten_call_graph_test.cc | 54 last_value = builder.AddInstruction(HloInstruction::CreateMap( in MakeMappingComputation()
|
| D | call_graph_test.cc | 56 last_value = builder.AddInstruction(HloInstruction::CreateMap( in MakeMappingComputation()
215 HloInstruction::CreateMap(kScalarShape, {call}, subcomputation)); in TEST_F()
|
| D | hlo_instruction_test.cc | 348 HloInstruction::CreateMap(f32a100x10, {param0}, add_f32)); in TEST_F()
769 HloInstruction::CreateMap(scalar_shape, {constant}, computation_x)); in TEST_F()
771 HloInstruction::CreateMap(scalar_shape, {map_1_x}, computation_x)); in TEST_F()
773 HloInstruction::CreateMap(scalar_shape, {map_2_x}, computation_y)); in TEST_F()
1030 HloInstruction::CreateMap(r2f32, {x}, map_computation)); in TEST_F()
|
| D | hlo_cse_test.cc | 637 HloInstruction::CreateMap(constant->shape(), {constant}, rng_function)); in TEST_F()
639 HloInstruction::CreateMap(constant->shape(), {constant}, rng_function)); in TEST_F()
|
| D | hlo_creation_utils.cc | 247 HloInstruction::CreateMap(map_shape, operands, map_computation)); in MakeMapHlo()
|
| D | hlo_instruction.h | 431 static std::unique_ptr<HloInstruction> CreateMap(
|
| D | buffer_assignment_test.cc | 754 HloInstruction::CreateMap(f32a100x10_, {param0}, map_computation)); in TEST_F()
1226 HloInstruction::CreateMap(vec_shape, {call}, map_computation)); in TEST_F()
|
| D | hlo_instruction.cc | 286 instruction = CreateMap(shape, all_operands(), computations(0)); in CreateFromProto()
774 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateMap( in CreateMap() function in xla::HloInstruction
|
| D | algebraic_simplifier.cc | 3073 new_reduce = computation_->AddInstruction(HloInstruction::CreateMap( in HandleReduce()
3098 HloInstruction::CreateMap(reduce_window->shape(), in HandleReduceWindow()
|
| D | hlo_parser.cc | 1190 HloInstruction::CreateMap(shape, operands, *to_apply)); in ParseInstructionRhs()
|
| D | algebraic_simplifier_test.cc | 492 builder.AddInstruction(HloInstruction::CreateMap( in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 514 HloInstruction::CreateMap(shape, {exp}, CreateAdderToOne(module.get()))); in TEST_F()
538 HloInstruction::CreateMap(shape, {exp0, exp1}, CreateMax(module.get()))); in TEST_F()
|