Searched defs:MatrixDescriptor (Results 1 – 2 of 2) sorted by relevance
73 struct MatrixDescriptor { struct74 se::DeviceMemoryBase data;75 se::blas::Transpose transpose;76 int64 num_rows;77 int64 num_cols;78 int64 stride;80 int64 reduced_dim() const { in reduced_dim()85 se::DeviceMemory<T> cast() const { in cast()
55 struct MatrixDescriptor { struct56 mlir::Value converted_value;57 bool transpose; // Whether this matrix needs to be transposed.58 int64_t num_rows;59 int64_t num_cols;