fix: Implement `aten.mean.default` and `aten.mean.dim` converters by gs-olive · Pull Request #1810 · pytorch/TensorRT

@gs-olive

- Replace existing implementation of aten.mean.dim with general version
invoking centralized `add_reduce_layer` utility
- Add implementation of aten.mean.default by refactoring default case to
be a special invocation of aten.mean.dim (specifically, one with `dim`
being all dimensions in the Tensor and `keepdim` being False).
- Add defaults for optional arguments in converters
- Add test cases for combinations of input options to the converters

narendasan

frank-wei

bowang007 pushed a commit that referenced this pull request

Apr 28, 2023

@gs-olive @bowang007