Skip to content

Questions around _aten_native_batch_norm_inference_onnx #786

New issue
New issue
Closed
Closed
Questions around _aten_native_batch_norm_inference_onnx#786
Assignees
xiaowuhufatcat-z
Labels
category: questionA question about somethingmodule: torchlibRelated to the torch/aten function lib in development
@abock

Description

@abock
abock
opened on Jun 15, 2023

As commented in #773, @gramalingam is curious what the intent of these lines are:

onnxscript/onnxscript/function_libs/torch_lib/ops/core.py

Lines 4435 to 4438 in f106c1e

# Cannot return 2 dup output, so have to do twice with different variable name
empty_mean = op.Cast(op.Shape(input, start=0, end=0), to=FLOAT.dtype)
empty_var = op.Cast(op.Shape(input, start=0, end=0), to=FLOAT.dtype)
return norm, empty_mean, empty_var

Metadata

Metadata

Assignees

Labels

category: questionA question about somethingmodule: 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