Searched defs:MatrixLayout (Results 1 – 6 of 6) sorted by relevance
51 struct MatrixLayout { struct58 static StatusOr<MatrixLayout> For(const Shape& shape); argument73 PrimitiveType dtype;77 int64_t num_rows;101 MatrixLayout lhs_layout; argument
66 explicit MatrixLayout(cublasLtMatrixLayout_t handle) in MatrixLayout() function
51 enum MatrixLayout enum
38 enum MatrixLayout { kRowMajor, kColumnMajor }; enum