Searched refs:matrix8_4 (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 1218 const Shape matrix8_4 = ShapeUtil::MakeShape(F32, {8, 4}); in TEST_F() local
1223 HloOpcode::kAdd, cube, matrix8_4, {1, 2}); in TEST_F()
1243 const Shape matrix8_4 = ShapeUtil::MakeShape(F32, {8, 4}); in TEST_F() local
1269 HloOpcode::kAdd, tensor, matrix8_4, {0, 1, 2}); in TEST_F()
1276 HloOpcode::kAdd, tensor, matrix8_4, {3, 0}); in TEST_F()
1283 HloOpcode::kAdd, tensor, matrix8_4, {2, 1}); in TEST_F()
|