Skip to content

Commit d0a6f57

Browse files
authored
Add reduce kernels for bigger types (#21490)
1 parent 8c2ee7b commit d0a6f57

File tree

4 files changed

+187
-22
lines changed

4 files changed

+187
-22
lines changed

docs/OperatorKernels.md

+20-20
Original file line numberDiff line numberDiff line change
@@ -272,18 +272,18 @@ Do not modify directly.*
272272
|Range|*in* start:**T**<br> *in* limit:**T**<br> *in* delta:**T**<br> *out* output:**T**|11+|**T** = tensor(double), tensor(float), tensor(int16), tensor(int32), tensor(int64)|
273273
|Reciprocal|*in* X:**T**<br> *out* Y:**T**|13+|**T** = tensor(double), tensor(float)|
274274
|||[6, 12]|**T** = tensor(double), tensor(float)|
275-
|ReduceL1|*in* data:**T**<br> *in* axes:**tensor(int64)**<br> *out* reduced:**T**<br><br>or<br><br>*in* data:**T**<br> *out* reduced:**T**|18+|**T** = tensor(float), tensor(int32), tensor(int64)|
276-
|||[13, 17]|**T** = tensor(float), tensor(int32), tensor(int64)|
277-
|||[11, 12]|**T** = tensor(float), tensor(int32), tensor(int64)|
278-
|||[1, 10]|**T** = tensor(float), tensor(int32), tensor(int64)|
279-
|ReduceL2|*in* data:**T**<br> *in* axes:**tensor(int64)**<br> *out* reduced:**T**<br><br>or<br><br>*in* data:**T**<br> *out* reduced:**T**|18+|**T** = tensor(float), tensor(int32), tensor(int64)|
280-
|||[13, 17]|**T** = tensor(float), tensor(int32), tensor(int64)|
281-
|||[11, 12]|**T** = tensor(float), tensor(int32), tensor(int64)|
282-
|||[1, 10]|**T** = tensor(float), tensor(int32), tensor(int64)|
283-
|ReduceLogSum|*in* data:**T**<br> *in* axes:**tensor(int64)**<br> *out* reduced:**T**<br><br>or<br><br>*in* data:**T**<br> *out* reduced:**T**|18+|**T** = tensor(float), tensor(int32), tensor(int64)|
284-
|||[13, 17]|**T** = tensor(float), tensor(int32), tensor(int64)|
285-
|||[11, 12]|**T** = tensor(float), tensor(int32), tensor(int64)|
286-
|||[1, 10]|**T** = tensor(float), tensor(int32), tensor(int64)|
275+
|ReduceL1|*in* data:**T**<br> *in* axes:**tensor(int64)**<br> *out* reduced:**T**<br><br>or<br><br>*in* data:**T**<br> *out* reduced:**T**|18+|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
276+
|||[13, 17]|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
277+
|||[11, 12]|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
278+
|||[1, 10]|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
279+
|ReduceL2|*in* data:**T**<br> *in* axes:**tensor(int64)**<br> *out* reduced:**T**<br><br>or<br><br>*in* data:**T**<br> *out* reduced:**T**|18+|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
280+
|||[13, 17]|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
281+
|||[11, 12]|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
282+
|||[1, 10]|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
283+
|ReduceLogSum|*in* data:**T**<br> *in* axes:**tensor(int64)**<br> *out* reduced:**T**<br><br>or<br><br>*in* data:**T**<br> *out* reduced:**T**|18+|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
284+
|||[13, 17]|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
285+
|||[11, 12]|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
286+
|||[1, 10]|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
287287
|ReduceLogSumExp|*in* data:**T**<br> *in* axes:**tensor(int64)**<br> *out* reduced:**T**<br><br>or<br><br>*in* data:**T**<br> *out* reduced:**T**|18+|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
288288
|||[13, 17]|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
289289
|||[11, 12]|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
@@ -294,20 +294,20 @@ Do not modify directly.*
294294
|||12|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64), tensor(int8), tensor(uint8)|
295295
|||11|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
296296
|||[1, 10]|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
297-
|ReduceMean|*in* data:**T**<br> *in* axes:**tensor(int64)**<br> *out* reduced:**T**<br><br>or<br><br>*in* data:**T**<br> *out* reduced:**T**|18+|**T** = tensor(double), tensor(float), tensor(int32)|
298-
|||[13, 17]|**T** = tensor(double), tensor(float), tensor(int32)|
299-
|||[11, 12]|**T** = tensor(double), tensor(float), tensor(int32)|
300-
|||[1, 10]|**T** = tensor(double), tensor(float), tensor(int32)|
297+
|ReduceMean|*in* data:**T**<br> *in* axes:**tensor(int64)**<br> *out* reduced:**T**<br><br>or<br><br>*in* data:**T**<br> *out* reduced:**T**|18+|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
298+
|||[13, 17]|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
299+
|||[11, 12]|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
300+
|||[1, 10]|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
301301
|ReduceMin|*in* data:**T**<br> *in* axes:**tensor(int64)**<br> *out* reduced:**T**<br><br>or<br><br>*in* data:**T**<br> *out* reduced:**T**|20+|**T** = tensor(bool), tensor(double), tensor(float), tensor(int32), tensor(int64), tensor(int8), tensor(uint8)|
302302
|||[18, 19]|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64), tensor(int8), tensor(uint8)|
303303
|||[13, 17]|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64), tensor(int8), tensor(uint8)|
304304
|||12|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64), tensor(int8), tensor(uint8)|
305305
|||11|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
306306
|||[1, 10]|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
307-
|ReduceProd|*in* data:**T**<br> *in* axes:**tensor(int64)**<br> *out* reduced:**T**<br><br>or<br><br>*in* data:**T**<br> *out* reduced:**T**|18+|**T** = tensor(float), tensor(int32), tensor(int64)|
308-
|||[13, 17]|**T** = tensor(float), tensor(int32), tensor(int64)|
309-
|||[11, 12]|**T** = tensor(float), tensor(int32), tensor(int64)|
310-
|||[1, 10]|**T** = tensor(float), tensor(int32), tensor(int64)|
307+
|ReduceProd|*in* data:**T**<br> *in* axes:**tensor(int64)**<br> *out* reduced:**T**<br><br>or<br><br>*in* data:**T**<br> *out* reduced:**T**|18+|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
308+
|||[13, 17]|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
309+
|||[11, 12]|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
310+
|||[1, 10]|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
311311
|ReduceSum|*in* data:**T**<br> *in* axes:**tensor(int64)**<br> *out* reduced:**T**<br><br>or<br><br>*in* data:**T**<br> *out* reduced:**T**|13+|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
312312
|||[11, 12]|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|
313313
|||[1, 10]|**T** = tensor(double), tensor(float), tensor(int32), tensor(int64)|

0 commit comments

Comments
 (0)