Lines Matching defs:matrix

103 // Return true if the given map represents a transposed matrix load,
144 // Return true if the transfer op can be converted to a MMA matrix load.
168 // Return true if the transfer op can be converted to a MMA matrix store.
187 /// converted to a MMA constant matrix op.
328     // converted to matrix type.
365     // If any instruction cannot use MMA matrix type drop the whole
366     // chain. MMA matrix are stored in an opaque type so they cannot be used
613   Value matrix = it->second;
615       op.getLoc(), matrix, op.getSource(), op.getIndices(),
626 /// Returns the vector type which represents a matrix fragment.
672 /// Check if the loaded matrix operand requires transposed.
688   // Output 2D matrix dimensions in the order of d0, d1.
909   Value matrix = it->second;
933         rewriter.create<vector::ExtractOp>(loc, matrix, ArrayRef<int64_t>{i});
1086   auto matrix = rewriter.create<gpu::SubgroupMmaConstantMatrixOp>(
1088   valueMapping[op.getResult()] = matrix;
1105   auto matrix = rewriter.create<gpu::SubgroupMmaConstantMatrixOp>(
1107   valueMapping[op.getResult()] = matrix;
1201 /// Convert an elementwise op to the equivalent elementwise op on MMA matrix.