Searched refs:r3f32 (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 606 Shape r3f32 = ShapeUtil::MakeShape(F32, {5, 6, 7}); in TEST_F() local
608 HloInstruction::CreateReduce(r3f32, param, zero, dims0, add_computation)); in TEST_F()
2237 Shape r3f32 = in TEST_F() local
2241 HloInstruction::CreateParameter(0, r3f32, "param0")); in TEST_F()
2243 HloInstruction::CreateParameter(1, r3f32, "param1")); in TEST_F()
2245 HloInstruction::CreateParameter(2, r3f32, "param2")); in TEST_F()
4676 Shape r3f32 = ShapeUtil::MakeShape(F32, {2, 2, 2}); in TEST_F() local
4678 HloInstruction::CreateBroadcast(r3f32, inner_bcast, {0, 2})); in TEST_F()
4695 Shape r3f32 = ShapeUtil::MakeShape(F32, {2, 5, 3}); in TEST_F() local
4701 HloInstruction::CreateBroadcast(r3f32, param0, {0, 2})); in TEST_F()
[all …]
|