rml_op_gemm_params
Parameters for the RML_OP_GEMM operation.
Y = alpha * transA(A) * transB(B) + beta * C
API
Graph Manipulation
Items
Name |
Type |
Description |
---|---|---|
input_a |
Operation that outputs tensor containing input A. |
|
input_b |
Operation that outputs tensor containing input B. |
|
input_c |
Operation that outputs tensor containing input C, optional. |
|
alpha |
float |
A scalar multiplier for the product of inputs A and B. |
beta |
float |
A scalar multiplier for the optional input C. |
transpose_a |
A transform to be applied to the A tensor; either a transpose, or no transform. |
|
transpose_b |
A transform to be applied to the B tensor; either a transpose, or no transform. |