Home
last modified time | relevance | path

Searched refs:matrix_32_64_ (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc48 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 …]