| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_memory_scheduler_test.cc | 219 const Shape r1f32 = ShapeUtil::MakeShape(xla::F32, {6}); in TEST_F() local
226 HloInstruction::CreateUnary(r1f32, HloOpcode::kAbs, lit)); in TEST_F()
229 HloInstruction::CreateUnary(r1f32, HloOpcode::kAbs, abs_const)); in TEST_F()
233 HloInstruction::CreateGetTupleElement(r1f32, tuple, 0)); in TEST_F()
236 HloInstruction::CreateUnary(r1f32, HloOpcode::kAbs, abs_const)); in TEST_F()
238 builder.AddInstruction(HloInstruction::CreateBinary(r1f32, HloOpcode::kAdd, in TEST_F()
263 const Shape r1f32 = ShapeUtil::MakeShape(xla::F32, {5}); in TEST_F() local
274 HloInstruction::CreateBinary(r1f32, HloOpcode::kAdd, c1, c2)); in TEST_F()
276 HloInstruction::CreateBinary(r1f32, HloOpcode::kMultiply, add, c3)); in TEST_F()
280 HloInstruction::CreateGetTupleElement(r1f32, tuple, 0)); in TEST_F()
[all …]
|
| D | hlo_instruction_test.cc | 1063 const Shape r1f32 = ShapeUtil::MakeShape(F32, {5}); in TEST_F() local
1066 builder.AddInstruction(HloInstruction::CreateParameter(0, r1f32, "x")); in TEST_F()
1068 builder.AddInstruction(HloInstruction::CreateParameter(1, r1f32, "y")); in TEST_F()
1070 HloInstruction::CreateBinary(r1f32, HloOpcode::kAdd, x, y)); in TEST_F()
1098 const Shape r1f32 = ShapeUtil::MakeShape(F32, {5}); in TEST_F() local
1121 builder.AddInstruction(HloInstruction::CreateParameter(3, r1f32, "p3")); in TEST_F()
1159 const Shape r1f32 = ShapeUtil::MakeShape(F32, {5}); in TEST_F() local
1163 builder.AddInstruction(HloInstruction::CreateParameter(0, r1f32, "x")); in TEST_F()
1167 builder.AddInstruction(HloInstruction::CreateBroadcast(r1f32, y, {})); in TEST_F()
1169 HloInstruction::CreateBinary(r1f32, HloOpcode::kMinimum, x, broadcast)); in TEST_F()
[all …]
|
| D | algebraic_simplifier_test.cc | 610 Shape r1f32 = ShapeUtil::MakeShape(F32, {5}); in TEST_F() local
611 builder.AddInstruction(HloInstruction::CreateReduce(r1f32, reduce0, zero, in TEST_F()
1082 Shape r1f32 = ShapeUtil::MakeShape(F32, {7}); in TEST_F() local
1085 HloInstruction::CreateParameter(0, r1f32, "param0")); in TEST_F()
1087 HloInstruction::CreateParameter(1, r1f32, "param1")); in TEST_F()
1089 HloInstruction::CreateParameter(2, r1f32, "param2")); in TEST_F()
1091 HloInstruction::CreateBinary(r1f32, HloOpcode::kPower, param1, param2)); in TEST_F()
1093 HloInstruction::CreateBinary(r1f32, HloOpcode::kDivide, param0, power)); in TEST_F()
1114 Shape r1f32 = ShapeUtil::MakeShape(F32, {3}); in TEST_F() local
1117 HloInstruction::CreateParameter(0, r1f32, "param0")); in TEST_F()
[all …]
|
| D | heap_simulator_test.cc | 133 const Shape r1f32 = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
140 HloInstruction::CreateParameter(0, r1f32, "cond_param")); in TEST_F()
157 HloInstruction::CreateParameter(0, r1f32, "body_param")); in TEST_F()
163 r1f32, HloOpcode::kSubtract, body_param, one_vector)); in TEST_F()
174 r1f32, cond_computation, body_computation, while_init)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | concat_test.cc | 415 auto r1f32 = xla::ShapeUtil::MakeShape(xla::F32, {1}); in XLA_TEST_F() local
416 auto x = Parameter(&builder, 0, r1f32, "x"); in XLA_TEST_F()
430 auto r1f32 = xla::ShapeUtil::MakeShape(xla::F32, {1}); in XLA_TEST_F() local
431 auto x = Parameter(&builder, 0, r1f32, "x"); in XLA_TEST_F()
|