Skip to content

[torchlib] native_batch_norm needs more aten ops support #1140

@titaiwangms

Description

@titaiwangms
  1. func: _native_batch_norm_legit.no_stats(Tensor input, Tensor? weight, Tensor? bias, bool training, float momentum, float eps) -> (Tensor, Tensor, Tensor)
  2. _native_batch_norm_legit_functional (generated by functionalization pass)
  3. aten::to_copy_.default

Metadata

Metadata

Assignees

Labels

module: torchlibRelated to the torch/aten function lib in development

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions