@@ -1523,9 +1523,9 @@ def dot(input, other, acc=None, input_precision=None, allow_tf32=None, max_num_i
15231523 where the first dimension of each block represents the batch dimension.
15241524
15251525 :param input: The first tensor to be multiplied.
1526- :type input: 2D or 3D tensor of scalar-type in {:code:`int8`, :code: `float8_e5m2`, :code:`float16`, :code:`bfloat16`, :code:`float32`}
1526+ :type input: 2D or 3D tensor of scalar-type in {:code:`int8`, :code:`float8_e5m2`, :code:`float16`, :code:`bfloat16`, :code:`float32`}
15271527 :param other: The second tensor to be multiplied.
1528- :type other: 2D or 3D tensor of scalar-type in {:code:`int8`, :code: `float8_e5m2`, :code:`float16`, :code:`bfloat16`, :code:`float32`}
1528+ :type other: 2D or 3D tensor of scalar-type in {:code:`int8`, :code:`float8_e5m2`, :code:`float16`, :code:`bfloat16`, :code:`float32`}
15291529 :param acc: The accumulator tensor. If not None, the result is added to this tensor.
15301530 :type acc: 2D or 3D tensor of scalar-type in {:code:`float16`, :code:`float32`, :code:`int32`}
15311531 :param input_precision: How to exercise the Tensor Cores for f32 x f32. If
@@ -1559,13 +1559,13 @@ def dot_scaled(lhs, lhs_scale, lhs_format, rhs, rhs_scale, rhs_format, acc=None,
15591559 :type lhs: 2D tensor representing fp4 or fp8 elements packed into uint8 for fp4 inputs, or in uint8 or the corresponding fp8 type for fp8 inputs.
15601560 :param lhs_scale: Scale factor for lhs tensor.
15611561 :type lhs_scale: e8m0 type represented as an uint8 tensor.
1562- :param lhs_format: format of the lhs tensor. Available formats: {:code:`e2m1`, :code:`e4m3`, :code: `e5m2`}.
1562+ :param lhs_format: format of the lhs tensor. Available formats: {:code:`e2m1`, :code:`e4m3`, :code:`e5m2`}.
15631563 :type lhs_format: str
15641564 :param rhs: The second tensor to be multiplied.
15651565 :type rhs: 2D tensor representing fp8 or bf16 elements in uint8 or the corresponding fp8 type for fp8 inputs or bf16 for bf16 inputs.
15661566 :param rhs_scale: Scale factor for rhs tensor.
15671567 :type rhs_scale: e8m0 type represented as an uint8 tensor.
1568- :param rhs_format: format of the rhs tensor. Available formats: {:code:`e4m3`, :code: `e5m2`, :code:`bf16`}.
1568+ :param rhs_format: format of the rhs tensor. Available formats: {:code:`e4m3`, :code:`e5m2`, :code:`bf16`}.
15691569 :type rhs_format: str
15701570 :param acc: The accumulator tensor. If not None, the result is added to this tensor.
15711571 """
0 commit comments