| /external/tensorflow/tensorflow/compiler/xla/service/ |
| map_inliner_test.cc | 59 HloInstruction::CreateMap(lhs->shape(), {lhs, rhs}, max_f32.get()));
93 HloInstruction::CreateMap(lhs->shape(), {lhs}, const2_f32.get()));
131 HloInstruction::CreateMap(lhs->shape(), {lhs, rhs}, max_f32.get()));
163 HloInstruction::CreateMap(lhs->shape(), {lhs, rhs}, param_f32.get()));
|
| flatten_call_graph_test.cc | 54 last_value = builder.AddInstruction(HloInstruction::CreateMap(
|
| hlo_computation_test.cc | 67 HloInstruction::CreateMap(r0f32_, {param}, map_computation));
107 HloInstruction::CreateMap(r0f32_, {param}, map1_computation));
109 HloInstruction::CreateMap(r0f32_, {param}, map2_computation));
|
| call_graph_test.cc | 56 last_value = builder.AddInstruction(HloInstruction::CreateMap(
215 HloInstruction::CreateMap(kScalarShape, {call}, subcomputation));
|
| hlo_cse_test.cc | 637 HloInstruction::CreateMap(constant->shape(), {constant}, rng_function));
639 HloInstruction::CreateMap(constant->shape(), {constant}, rng_function));
|
| hlo_instruction_test.cc | 348 HloInstruction::CreateMap(f32a100x10, {param0}, add_f32));
769 HloInstruction::CreateMap(scalar_shape, {constant}, computation_x));
771 HloInstruction::CreateMap(scalar_shape, {map_1_x}, computation_x));
773 HloInstruction::CreateMap(scalar_shape, {map_2_x}, computation_y));
[all...] |
| hlo_creation_utils.cc | 247 HloInstruction::CreateMap(map_shape, operands, map_computation));
|
| hlo_instruction.h | 431 static std::unique_ptr<HloInstruction> CreateMap(
[all...] |
| algebraic_simplifier.cc | [all...] |
| buffer_assignment_test.cc | [all...] |
| hlo_instruction.cc | 286 instruction = CreateMap(shape, all_operands(), computations(0));
774 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateMap(
[all...] |
| hlo_parser.cc | [all...] |
| algebraic_simplifier_test.cc | 492 builder.AddInstruction(HloInstruction::CreateMap(
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_instruction_fusion_test.cc | 514 HloInstruction::CreateMap(shape, {exp}, CreateAdderToOne(module.get())));
538 HloInstruction::CreateMap(shape, {exp0, exp1}, CreateMax(module.get())));
[all...] |
