| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_cost_analysis_test.cc | 694 Shape r2f32 = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
715 HloInstruction::CreateBinary(r2f32, HloOpcode::kAdd, c1, c2)); in TEST_F()
717 HloInstruction::CreateTernary(r2f32, HloOpcode::kClamp, c2, add, add)); in TEST_F()
719 HloInstruction::CreateUnary(r2f32, HloOpcode::kExp, add)); in TEST_F()
721 HloInstruction::CreateBinary(r2f32, HloOpcode::kMultiply, exp, c3)); in TEST_F()
723 HloInstruction::CreateBinary(r2f32, HloOpcode::kSubtract, mul, clamp)); in TEST_F()
761 Shape r2f32 = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
776 HloInstruction::CreateBinary(r2f32, HloOpcode::kAdd, c1, c2)); in TEST_F()
778 HloInstruction::CreateTernary(r2f32, HloOpcode::kClamp, c2, add, add)); in TEST_F()
780 HloInstruction::CreateUnary(r2f32, HloOpcode::kExp, add)); in TEST_F()
[all …]
|
| D | algebraic_simplifier_test.cc | 722 Shape r2f32 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST_F() local
725 HloInstruction::CreateParameter(0, r2f32, "param0")); in TEST_F()
729 HloInstruction::CreateBroadcast(r2f32, zero, {0, 1})); in TEST_F()
731 HloInstruction::CreateBinary(r2f32, HloOpcode::kAdd, bcast, param0)); in TEST_F()
758 Shape r2f32 = ShapeUtil::MakeShape(F32, {32, 1}); in TEST_F() local
760 HloInstruction::CreateParameter(0, r2f32, "param0")); in TEST_F()
764 r2f32, in TEST_F()
766 HloInstruction::CreateBroadcast(r2f32, zero, {}))}, in TEST_F()
805 Shape r2f32 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST_F() local
808 HloInstruction::CreateParameter(0, r2f32, "param0")); in TEST_F()
[all …]
|
| D | hlo_instruction_test.cc | 1181 const Shape r2f32 = ShapeUtil::MakeShapeWithLayout(F32, {10, 10}, {1, 0}); in TEST_F() local
1188 builder.AddInstruction(HloInstruction::CreateParameter(0, r2f32, "x")); in TEST_F()
1190 HloInstruction::CreateMap(r2f32, {x}, map_computation)); in TEST_F()
1198 const Shape r2f32 = ShapeUtil::MakeShape(F32, {3, 5}); in TEST_F() local
1214 builder.AddInstruction(HloInstruction::CreateParameter(0, r2f32, "p0")); in TEST_F()
1216 builder.AddInstruction(HloInstruction::CreateParameter(1, r2f32, "p1")); in TEST_F()
1218 builder.AddInstruction(HloInstruction::CreateParameter(2, r2f32, "p2")); in TEST_F()
1222 HloInstruction::CreateBinary(r2f32, HloOpcode::kMultiply, p0, p1)); in TEST_F()
1224 HloInstruction::CreateBinary(r2f32, HloOpcode::kDivide, mul, p2)); in TEST_F()
1227 builder.AddInstruction(HloInstruction::CreateBroadcast(r2f32, p3, {1})); in TEST_F()
[all …]
|
| D | heap_simulator_test.cc | 133 const Shape r2f32 = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F() local
177 HloInstruction::CreateBroadcast(r2f32, while_loop, {1})); in TEST_F()
184 HloInstruction::CreateTranspose(r2f32, matrix, {0, 1})); in TEST_F()
188 HloInstruction::CreateBinary(r2f32, HloOpcode::kAdd, transpose, bcast)); in TEST_F()
|