Visible to Intel only — GUID: GUID-23A1D718-0F24-48CC-BD7C-610E4037AE0E
Visible to Intel only — GUID: GUID-23A1D718-0F24-48CC-BD7C-610E4037AE0E
ReduceProd
General
ReduceProd operation performs the reduction with multiplication on a given src data along dimensions specified by axes.
Take channel axis = 0 and keep_dims = True as an example:
Operation attributes
Attribute Name |
Description |
Value Type |
Supported Values |
Required or Optional |
---|---|---|---|---|
Specify indices of src tensor, along which the reduction is performed. If axes is a list, reduce over all of them. If axes is empty, corresponds to the identity operation. If axes contains all dimensions of src tensor, a single reduction value is calculated for the entire src tensor. Exactly one of attribute axes and the second input tensor axes should be available. |
s64 |
A s64 list values which is in the range of [-r, r-1] where r = rank(src). Empty list(default) |
Optional |
|
If set to true it holds axes that are used for reduction. For each such axes, dst dimension is equal to 1. |
bool |
true , false (default) |
Optional |
Execution arguments
The inputs and outputs must be provided according to below index order when constructing an operation.
Inputs
Index |
Argument Name |
Required or Optional |
---|---|---|
0 |
src |
Required |
1 |
axes |
Optional |
Outputs
Index |
Argument Name |
Required or Optional |
---|---|---|
0 |
dst |
Required |
Supported data types
ReduceProd operation supports the following data type combinations.
Source/Destination |
Axes |
---|---|
f32 |
s32 |
bf16 |
s32 |
f16 |
s32 |