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

rml_op

Operation that outputs tensor containing input A.

input_b

rml_op

Operation that outputs tensor containing input B.

input_c

rml_op

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

rml_bool

A transform to be applied to the A tensor; either a transpose, or no transform.

transpose_b

rml_bool

A transform to be applied to the B tensor; either a transpose, or no transform.