diff options
| author | Zhi An Ng <zhin@google.com> | 2022-08-24 18:52:39 -0700 |
|---|---|---|
| committer | XNNPACK Team <xnnpack-github-robot@google.com> | 2022-08-24 18:53:35 -0700 |
| commit | 360bdd17dce31968bea7417b0d75837e97cd30c3 (patch) | |
| tree | e65e1c3e70614159a99f37b5ff16dc8750a5e21b | |
| parent | fee9675af8160c6d0474c87174785b586121b151 (diff) | |
| download | XNNPACK-360bdd17dce31968bea7417b0d75837e97cd30c3.tar.gz | |
Update xnnpack.h to state that convolution 2d node supports not having a bias.
PiperOrigin-RevId: 469878509
| -rw-r--r-- | include/xnnpack.h | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/include/xnnpack.h b/include/xnnpack.h index 5414d8ffa..8825efc4c 100644 --- a/include/xnnpack.h +++ b/include/xnnpack.h @@ -337,8 +337,9 @@ enum xnn_status xnn_define_convert( /// @param filter_id - Value ID for the filter tensor. The filter tensor must ge a 4D tensor defined in the @a subgraph /// with [groups * group_output_channels, kernel_height, kernel_width, group_input_channels] /// dimensions. -/// @param bias_id - Value ID for the bias tensor. The bias tensor must be a 1D tensor defined in the @a subgraph with -/// [groups * group_output_channels] dimensions. +/// @param bias_id - Value ID for the bias tensor, or XNN_INVALID_VALUE_ID for a 2D Convolution Node without a bias. If +/// present, the bias tensor must be a 1D tensor defined in the @a subgraph with [groups * +/// group_output_channels] dimensions. /// @param output_id - Value ID for the output tensor. The output tensor must be a 4D tensor defined in the @a subgraph /// with [N, OH, OW, groups * group_output_channels] dimensions. /// @param flags - binary features of the 2D Convolution Node. The only currently supported values is |