Skip to content

Test torchscript graph building with wrapped tensors in trace_only function #1917

@titaiwangms

Description

@titaiwangms

The trace_only function was designed to allow flexible writing, so it does not wrap python constants until it hits an ONNX op. However, this behaviorbecomes complicated when it comes to dynamic shapes, as sym_size_int is both scalar and tensor.

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions