Closed
Description
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