reducer

torch_to_nnef.op.aten.reducer

argmax

argmax(node, op_helper, **kwargs)

Map PyTorch: 'aten:argmax' to NNEF.

argmin

argmin(node, op_helper, **kwargs)

Map PyTorch: 'aten:argmin' to NNEF.

max_

max_(node, op_helper, **kwargs)

Map PyTorch: 'aten:max' to NNEF.

mean

mean(node, op_helper, **kwargs)

Map PyTorch: 'aten:mean' to NNEF.

min_

min_(node, op_helper, **kwargs)

Map PyTorch: 'aten:min' to NNEF.

prod

prod(node, op_helper, inference_target, **kwargs)

Map PyTorch: 'aten:prod' to NNEF.

reduce_all

reduce_all(node, op_helper, **kwargs)

Map PyTorch: 'aten:reduce_all', 'aten:all' to NNEF.

reduce_any

reduce_any(node, op_helper, **kwargs)

Map PyTorch: 'aten:reduce_any', 'aten:any' to NNEF.

reduce_max

reduce_max(node, op_helper, **kwargs)

Map PyTorch: 'aten:reduce_max', 'aten:amax' to NNEF.

reduce_min

reduce_min(node, op_helper, **kwargs)

Map PyTorch: 'aten:reduce_min', 'aten:amin' to NNEF.

reduce_sum

reduce_sum(node, op_helper, **kwargs)

Map PyTorch: 'aten:reduce_sum', 'aten:sum' to NNEF.