Searched refs:matrix_32_64_ (Results 1 – 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | shape_inference_test.cc | 48 const Shape matrix_32_64_ = ShapeUtil::MakeShape(F32, {32, 64}); member in xla::__anon8d0540a50111::ShapeInferenceTest 136 HloOpcode::kSelect, pred_, matrix_64_48_, matrix_32_64_); in TEST_F() 279 complex_shape(vector_64_, matrix_32_64_, {1})); in TEST_F() 282 complex_shape(matrix_32_64_, vector_64_, {1})); in TEST_F() 285 complex_shape(matrix_32_64_, matrix_32_64_, {})); in TEST_F() 287 TF_ASSERT_OK_AND_ASSIGN(result, complex_shape(matrix_32_64_, f32_, {})); in TEST_F() 981 ShapeUtil::MakeShape(F32, {32, 32, 32}), matrix_32_64_, dot_dnums); in TEST_F() 1007 ShapeInference::InferDotOpShape(matrix_32_64_, vector_64_, dot_dnums); in TEST_F() 1011 ShapeInference::InferDotOpShape(matrix_32_64_, vector_32_, dot_dnums); in TEST_F() 1021 ShapeInference::InferDotOpShape(vector_32_, matrix_32_64_, dot_dnums); in TEST_F() [all …]
|