Skip to content

BatchToSpaceND: support 3d inputs #572

@vadimkantorov

Description

@vadimkantorov

I'm trying to convert the OpenSeq2Seq's wav2letter model checkpoints from TensorFlow to ONNX.

It contains BatchToSpaceND ops. Inputs are however a 3d tensor. Currently tfonnx supports NHWC.

Proposal: if the input is 3d, insert automatically an Unsqueeze op or a rewriter in order to reshape to NH1C and then back to NHC.

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions