BNNSDirectApplyBroadcastMatMul(_:_:_:_:_:_:_:)
Applies a broadcast matrix multiplication operation directly to two input matrices.
Declaration
func BNNSDirectApplyBroadcastMatMul(_ transA: Bool, _ transB: Bool, _ alpha: Float, _ inputA: UnsafePointer<BNNSNDArrayDescriptor>, _ inputB: UnsafePointer<BNNSNDArrayDescriptor>, _ output: UnsafePointer<BNNSNDArrayDescriptor>, _ filter_params: UnsafePointer<BNNSFilterParameters>?)Parameters
- transA:
A Boolean value that transposes the last two dimensions of matrix A.
- transB:
A Boolean value that transposes the last two dimensions of matrix B.
- alpha:
A value to scale the result.
- inputA:
The descriptor of matrix A.
- inputB:
The descriptor of matrix B.
- output:
The descriptor of the output.
- filter_params:
The filter runtime parameters.
See Also
Matrix multiplication
BNNS.BroadcastMatrixMultiplyLayerBNNSLayerParametersBroadcastMatMulBNNSFilterCreateLayerBroadcastMatMul(_:_:)BNNSMatMulWorkspaceSize(_:_:_:_:_:_:_:)BNNSMatMul(_:_:_:_:_:_:_:_:)applyMatrixMultiplication(inputA:transposed:inputB:transposed:output:alpha:workspace:filterParameters:)matrixMultiplicationWorkspaceSize(inputA:transposed:inputB:transposed:output:alpha:filterParameters:)