Searched refs:matrix_64_48_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 49 const Shape matrix_64_48_ = ShapeUtil::MakeShape(F32, {64, 48}); member in xla::__anon8d0540a50111::ShapeInferenceTest
121 HloOpcode::kSelect, pred_, matrix_64_48_, matrix_64_48_); in TEST_F()
123 ASSERT_TRUE(ShapeUtil::Equal(matrix_64_48_, inferred_status.ValueOrDie())); in TEST_F()
129 HloOpcode::kSelect, predarray, matrix_64_48_, matrix_64_48_); in TEST_F()
131 ASSERT_TRUE(ShapeUtil::Equal(matrix_64_48_, inferred_status.ValueOrDie())); in TEST_F()
136 HloOpcode::kSelect, pred_, matrix_64_48_, matrix_32_64_); in TEST_F()
142 HloOpcode::kSelect, s32_, matrix_64_48_, matrix_64_48_); in TEST_F()
148 HloOpcode::kSelect, ShapeUtil::MakeShape(PRED, {64}), matrix_64_48_, in TEST_F()
149 matrix_64_48_); in TEST_F()
166 HloOpcode::kClamp, matrix_64_48_, matrix_64_48_, matrix_64_48_); in TEST_F()
[all …]
|