Lines Matching defs:matrix
39 /// when the elementwise op directly supports with cooperative matrix type.
96 /// matrix ops.
115 /// Converts elementwise ops to SPIR-V cooperative matrix elementwise ops for
124 // All operands should be of cooperative matrix types.
139 /// Converts elementwise ops to SPIR-V cooperative matrix elementwise ops for
140 /// matrix times scalar case.
151 // All operands should be of cooperative matrix types.
165 Value matrix = nullptr;
168 matrix = adaptor.getOperands().back();
170 matrix = adaptor.getOperands().front();
173 if (!splat || !matrix)
176 // Constant MMA matrix ops are converted to `spirv.CompositeConstruct` ops.
191 op, coopType, ValueRange{matrix, scalar});