DotGeneral Operator — infer_types_dot_general • stablehlo

See https://openxla.org/stablehlo/spec#dot_general for details.

Usage

infer_types_dot_general(lhs, rhs, dot_dimension_numbers)

hlo_dot_general(lhs, rhs, contracting_dims, batching_dims = NULL)

Arguments

lhs, rhs, contracting_dims, batching_dims: (FuncValue)
dot_dimension_numbers: (DotDimensionNumbers)
The dot dimension number.

Value