Home | History | Annotate | Download | only in service

Lines Matching refs:complex_shape

257   auto complex_shape = [&](const Shape& lhs, const Shape& rhs,
263   ASSERT_FALSE(complex_shape(s32_, s32_, {}).ok());
264   ASSERT_FALSE(complex_shape(pred_, pred_, {}).ok());
266   ASSERT_FALSE(complex_shape(f32_, f64_, {}).ok());
268   ASSERT_FALSE(complex_shape(f64_, f64_, {}).ok());
271   TF_ASSERT_OK_AND_ASSIGN(Shape result, complex_shape(f32_, f32_, {}));
273   TF_ASSERT_OK_AND_ASSIGN(result, complex_shape(vector_32_, f32_, {}));
275   TF_ASSERT_OK_AND_ASSIGN(result, complex_shape(f32_, vector_32_, {}));
277   TF_ASSERT_OK_AND_ASSIGN(result, complex_shape(vector_32_, f32_, {}));
282                           complex_shape(vector_64_, matrix_32_64_, {1}));
285                           complex_shape(matrix_32_64_, vector_64_, {1}));
288                           complex_shape(matrix_32_64_, matrix_32_64_, {}));
290   TF_ASSERT_OK_AND_ASSIGN(result, complex_shape(matrix_32_64_, f32_, {}));