Skip to content

Unsupported ops TF 2.14.0: OnesLike not supported #2267

Closed
@Gerstenberger

Description

@Gerstenberger

New Operator

Describe the operator

With tensorflow==2.14.0 the conversion of op OnesLike yields the error

Tensorflow op [output/rec/dec_0_self_att_att/ones_like: OnesLike] is not supported

which works fine using tensorflow==2.13.1.

Do you know this operator be constructed using existing ONNX operators?

yes, similar to ZerosLike using ConstantOfShape

Is this operator used by any model currently? Which one?

yes, in our SelfAttention (used eg. in our Transformer models) implementation for masking.

Are you willing to contribute it? (Y/N)

yes

Notes

For some notes see issue rwth-i6/returnn#1455
The onnx package versions used

tf2onnx==1.15.1
onnx==1.15.0
onnxruntime==1.16.1

Metadata

Metadata

Assignees

No one assigned

    Labels

    unsupported opsIssues related to unsupported operators

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions