| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | map_inliner_test.cc | 41 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
45 HloInstruction::CreateParameter(0, r0f32, "x")); in TEST_F()
47 HloInstruction::CreateParameter(1, r0f32, "y")); in TEST_F()
78 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
82 HloInstruction::CreateParameter(0, r0f32, "x")); in TEST_F()
111 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
117 HloInstruction::CreateParameter(1, r0f32, "x")); in TEST_F()
119 HloInstruction::CreateParameter(0, r0f32, "y")); in TEST_F()
149 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
152 param_builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32, "p0")); in TEST_F()
[all …]
|
| D | hlo_reachability_test.cc | 102 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
109 r0f32, HloOpcode::kAdd, constant1, constant2)); in TEST_F()
111 HloInstruction::CreateUnary(r0f32, HloOpcode::kNegate, constant2)); in TEST_F()
113 HloInstruction::CreateUnary(r0f32, HloOpcode::kExp, negate)); in TEST_F()
115 HloInstruction::CreateBinary(r0f32, HloOpcode::kMultiply, add, exp)); in TEST_F()
117 HloInstruction::CreateUnary(r0f32, HloOpcode::kCopy, exp)); in TEST_F()
|
| D | algebraic_simplifier_test.cc | 61 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
64 HloInstruction::CreateParameter(0, r0f32, "param0")); in TEST_F()
68 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, param0, zero)); in TEST_F()
653 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
656 HloInstruction::CreateParameter(0, r0f32, "param0")); in TEST_F()
660 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, constant, param0)); in TEST_F()
674 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
677 HloInstruction::CreateParameter(0, r0f32, "param0")); in TEST_F()
684 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, param0, constant1)); in TEST_F()
686 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, add1, constant2)); in TEST_F()
[all …]
|
| D | call_inliner_test.cc | 59 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
61 HloInstruction::CreateCall(r0f32, {}, inner_computation)); in TEST_F()
|
| D | hlo_instruction_test.cc | 332 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
339 HloInstruction::CreateParameter(0, r0f32, "x")); in TEST_F()
343 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, param, value)); in TEST_F()
370 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
377 HloInstruction::CreateParameter(0, r0f32, "x")); in TEST_F()
379 HloInstruction::CreateParameter(1, r0f32, "y")); in TEST_F()
381 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, paramx, paramy)); in TEST_F()
1257 const Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1264 builder.AddInstruction(HloInstruction::CreateParameter(1, r0f32, "y")); in TEST_F()
|
| D | buffer_assignment_test.cc | 2510 Shape r0f32 = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
2515 HloInstruction::CreateParameter(0, r0f32, "param")); in TEST_F()
2519 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, param, constant1)); in TEST_F()
2528 HloInstruction::CreateCall(r0f32, {constant2}, sub_computation)); in TEST_F()
2530 HloInstruction::CreateCall(r0f32, {constant3}, sub_computation)); in TEST_F()
2532 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, call1, constant2)); in TEST_F()
2534 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, call2, add1)); in TEST_F()
|
| D | heap_simulator_test.cc | 131 const Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
144 cond_builder.AddInstruction(HloInstruction::CreateReshape(r0f32, slice)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_aot_test_helper.cc | 40 auto r0f32 = xla::ShapeUtil::MakeShape(xla::F32, {}); in Doubler() local
41 auto x = xla::Parameter(&builder, 0, r0f32, "x"); in Doubler()
56 auto r0f32 = xla::ShapeUtil::MakeShape(xla::F32, {}); in main() local
57 auto sum = CustomCall(&builder, "SumStructElements", {opaque_param}, r0f32); in main()
88 &computation, /*argument_layouts=*/{&opaque_shape}, &r0f32}; in main()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_fusion_test.cc | 157 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
160 r0f32, HloOpcode::kAdd, in TEST_F()
162 HloInstruction::CreateParameter(0, r0f32, "x")), in TEST_F()
164 HloInstruction::CreateParameter(1, r0f32, "y")))); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | instruction_fusion_test.cc | 588 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
591 HloInstruction::CreateParameter(0, r0f32, "param0")); in TEST_F()
596 HloInstruction::CreateBinary(r0f32, HloOpcode::kDivide, param0, one)); in TEST_F()
598 HloInstruction::CreateBinary(r0f32, HloOpcode::kRemainder, param0, one)); in TEST_F()
|