Home
last modified time | relevance | path

Searched refs:r2f32 (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis_test.cc440 Shape r2f32 = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F() local
461 HloInstruction::CreateBinary(r2f32, HloOpcode::kAdd, c1, c2)); in TEST_F()
463 HloInstruction::CreateTernary(r2f32, HloOpcode::kClamp, c2, add, add)); in TEST_F()
465 HloInstruction::CreateUnary(r2f32, HloOpcode::kExp, add)); in TEST_F()
467 HloInstruction::CreateBinary(r2f32, HloOpcode::kMultiply, exp, c3)); in TEST_F()
469 HloInstruction::CreateBinary(r2f32, HloOpcode::kSubtract, mul, clamp)); in TEST_F()
Dalgebraic_simplifier_test.cc451 Shape r2f32 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST_F() local
454 HloInstruction::CreateParameter(0, r2f32, "param0")); in TEST_F()
458 HloInstruction::CreateBroadcast(r2f32, zero, {0, 1})); in TEST_F()
460 HloInstruction::CreateBinary(r2f32, HloOpcode::kAdd, bcast, param0)); in TEST_F()
487 Shape r2f32 = ShapeUtil::MakeShape(F32, {32, 1}); in TEST_F() local
489 HloInstruction::CreateParameter(0, r2f32, "param0")); in TEST_F()
493 r2f32, in TEST_F()
495 HloInstruction::CreateBroadcast(r2f32, zero, {}))}, in TEST_F()
534 Shape r2f32 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST_F() local
537 HloInstruction::CreateParameter(0, r2f32, "param0")); in TEST_F()
[all …]
Dhlo_instruction_test.cc1021 const Shape r2f32 = ShapeUtil::MakeShapeWithLayout(F32, {10, 10}, {1, 0}); in TEST_F() local
1028 builder.AddInstruction(HloInstruction::CreateParameter(0, r2f32, "x")); in TEST_F()
1030 HloInstruction::CreateMap(r2f32, {x}, map_computation)); in TEST_F()
1038 const Shape r2f32 = ShapeUtil::MakeShape(F32, {3, 5}); in TEST_F() local
1054 builder.AddInstruction(HloInstruction::CreateParameter(0, r2f32, "p0")); in TEST_F()
1056 builder.AddInstruction(HloInstruction::CreateParameter(1, r2f32, "p1")); in TEST_F()
1058 builder.AddInstruction(HloInstruction::CreateParameter(2, r2f32, "p2")); in TEST_F()
1062 HloInstruction::CreateBinary(r2f32, HloOpcode::kMultiply, p0, p1)); in TEST_F()
1064 HloInstruction::CreateBinary(r2f32, HloOpcode::kDivide, mul, p2)); in TEST_F()
1067 builder.AddInstruction(HloInstruction::CreateBroadcast(r2f32, p3, {1})); in TEST_F()
[all …]
Dheap_simulator_test.cc133 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()