| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | map_inliner_test.cc | 42 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
46 HloInstruction::CreateParameter(0, r0f32, "x")); in TEST_F()
48 HloInstruction::CreateParameter(1, r0f32, "y")); in TEST_F()
79 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
83 HloInstruction::CreateParameter(0, r0f32, "x")); in TEST_F()
112 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
118 HloInstruction::CreateParameter(1, r0f32, "x")); in TEST_F()
120 HloInstruction::CreateParameter(0, r0f32, "y")); in TEST_F()
150 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
153 param_builder.AddInstruction(HloInstruction::CreateParameter(0, r0f32, "p0")); in TEST_F()
[all …]
|
| D | hlo_reachability_test.cc | 101 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
108 r0f32, HloOpcode::kAdd, constant1, constant2)); in TEST_F()
110 HloInstruction::CreateUnary(r0f32, HloOpcode::kNegate, constant2)); in TEST_F()
112 HloInstruction::CreateUnary(r0f32, HloOpcode::kExp, negate)); in TEST_F()
114 HloInstruction::CreateBinary(r0f32, HloOpcode::kMultiply, add, exp)); in TEST_F()
116 HloInstruction::CreateUnary(r0f32, HloOpcode::kCopy, exp)); in TEST_F()
|
| D | algebraic_simplifier_test.cc | 60 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
63 HloInstruction::CreateParameter(0, r0f32, "param0")); in TEST_F()
67 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, param0, zero)); in TEST_F()
624 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
627 HloInstruction::CreateParameter(0, r0f32, "param0")); in TEST_F()
631 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, constant, param0)); in TEST_F()
645 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
648 HloInstruction::CreateParameter(0, r0f32, "param0")); in TEST_F()
655 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, param0, constant1)); in TEST_F()
657 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, add1, constant2)); in TEST_F()
[all …]
|
| D | call_inliner_test.cc | 60 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
62 HloInstruction::CreateCall(r0f32, {}, inner_computation)); in TEST_F()
|
| D | hlo_instruction_test.cc | 331 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
338 HloInstruction::CreateParameter(0, r0f32, "x")); in TEST_F()
342 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, param, value)); in TEST_F()
369 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
376 HloInstruction::CreateParameter(0, r0f32, "x")); in TEST_F()
378 HloInstruction::CreateParameter(1, r0f32, "y")); in TEST_F()
380 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, paramx, paramy)); in TEST_F()
1158 const Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1165 builder.AddInstruction(HloInstruction::CreateParameter(1, r0f32, "y")); in TEST_F()
|
| D | buffer_assignment_test.cc | 2432 Shape r0f32 = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
2437 HloInstruction::CreateParameter(0, r0f32, "param")); in TEST_F()
2441 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, param, constant1)); in TEST_F()
2450 HloInstruction::CreateCall(r0f32, {constant2}, sub_computation)); in TEST_F()
2452 HloInstruction::CreateCall(r0f32, {constant3}, sub_computation)); in TEST_F()
2454 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, call1, constant2)); in TEST_F()
2456 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, call2, add1)); in TEST_F()
|
| D | heap_simulator_test.cc | 132 const Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
145 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 | 158 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
161 r0f32, HloOpcode::kAdd, in TEST_F()
163 HloInstruction::CreateParameter(0, r0f32, "x")), in TEST_F()
165 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()
|