Skip to content

Fix computing of output dims in im2sequence op. #9977

@wanghaoshuang

Description

@wanghaoshuang

The dim[0] of input in infer_shape should be set to -1 because we can't get the batch size in compiling time. So the computing of dims should be removed from infer_shape to Compute function in kernel.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions