Searched refs:r3f32 (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 385 Shape r3f32 = ShapeUtil::MakeShape(F32, {5, 6, 7}); in TEST_F() local
387 HloInstruction::CreateReduce(r3f32, param, zero, dims0, add_computation)); in TEST_F()
1710 Shape r3f32 = in TEST_F() local
1714 HloInstruction::CreateParameter(0, r3f32, "param0")); in TEST_F()
1716 HloInstruction::CreateParameter(1, r3f32, "param1")); in TEST_F()
1718 HloInstruction::CreateParameter(2, r3f32, "param2")); in TEST_F()
3883 Shape r3f32 = ShapeUtil::MakeShape(F32, {2, 2, 2}); in TEST_F() local
3885 HloInstruction::CreateBroadcast(r3f32, inner_bcast, {0, 2})); in TEST_F()
3902 Shape r3f32 = ShapeUtil::MakeShape(F32, {2, 5, 3}); in TEST_F() local
3908 HloInstruction::CreateBroadcast(r3f32, param0, {0, 2})); in TEST_F()
[all …]
|