| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_memory_scheduler_test.cc | 218 const Shape r1f32 = ShapeUtil::MakeShape(xla::F32, {6}); in TEST_F() local
225 HloInstruction::CreateUnary(r1f32, HloOpcode::kAbs, lit)); in TEST_F()
228 HloInstruction::CreateUnary(r1f32, HloOpcode::kAbs, abs_const)); in TEST_F()
232 HloInstruction::CreateGetTupleElement(r1f32, tuple, 0)); in TEST_F()
235 HloInstruction::CreateUnary(r1f32, HloOpcode::kAbs, abs_const)); in TEST_F()
237 builder.AddInstruction(HloInstruction::CreateBinary(r1f32, HloOpcode::kAdd, in TEST_F()
262 const Shape r1f32 = ShapeUtil::MakeShape(xla::F32, {5}); in TEST_F() local
273 HloInstruction::CreateBinary(r1f32, HloOpcode::kAdd, c1, c2)); in TEST_F()
275 HloInstruction::CreateBinary(r1f32, HloOpcode::kMultiply, add, c3)); in TEST_F()
279 HloInstruction::CreateGetTupleElement(r1f32, tuple, 0)); in TEST_F()
[all …]
|
| D | hlo_instruction_test.cc | 1162 const Shape r1f32 = ShapeUtil::MakeShape(F32, {5}); in TEST_F() local
1165 builder.AddInstruction(HloInstruction::CreateParameter(0, r1f32, "x")); in TEST_F()
1167 builder.AddInstruction(HloInstruction::CreateParameter(1, r1f32, "y")); in TEST_F()
1169 HloInstruction::CreateBinary(r1f32, HloOpcode::kAdd, x, y)); in TEST_F()
1197 const Shape r1f32 = ShapeUtil::MakeShape(F32, {5}); in TEST_F() local
1220 builder.AddInstruction(HloInstruction::CreateParameter(3, r1f32, "p3")); in TEST_F()
1258 const Shape r1f32 = ShapeUtil::MakeShape(F32, {5}); in TEST_F() local
1262 builder.AddInstruction(HloInstruction::CreateParameter(0, r1f32, "x")); in TEST_F()
1266 builder.AddInstruction(HloInstruction::CreateBroadcast(r1f32, y, {})); in TEST_F()
1268 HloInstruction::CreateBinary(r1f32, HloOpcode::kMinimum, x, broadcast)); in TEST_F()
[all …]
|
| D | algebraic_simplifier_test.cc | 639 Shape r1f32 = ShapeUtil::MakeShape(F32, {5}); in TEST_F() local
640 builder.AddInstruction(HloInstruction::CreateReduce(r1f32, reduce0, zero, in TEST_F()
1149 Shape r1f32 = ShapeUtil::MakeShape(F32, {7}); in TEST_F() local
1152 HloInstruction::CreateParameter(0, r1f32, "param0")); in TEST_F()
1154 HloInstruction::CreateParameter(1, r1f32, "param1")); in TEST_F()
1156 HloInstruction::CreateParameter(2, r1f32, "param2")); in TEST_F()
1158 HloInstruction::CreateBinary(r1f32, HloOpcode::kPower, param1, param2)); in TEST_F()
1160 HloInstruction::CreateBinary(r1f32, HloOpcode::kDivide, param0, power)); in TEST_F()
1181 Shape r1f32 = ShapeUtil::MakeShape(F32, {3}); in TEST_F() local
1184 HloInstruction::CreateParameter(0, r1f32, "param0")); in TEST_F()
[all …]
|
| D | heap_simulator_test.cc | 132 const Shape r1f32 = ShapeUtil::MakeShape(F32, {4}); in TEST_F() local
139 HloInstruction::CreateParameter(0, r1f32, "cond_param")); in TEST_F()
156 HloInstruction::CreateParameter(0, r1f32, "body_param")); in TEST_F()
162 r1f32, HloOpcode::kSubtract, body_param, one_vector)); in TEST_F()
173 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()
|