-
Notifications
You must be signed in to change notification settings - Fork 3.6k
Description
Describe the bug
in GRU case,with the same parameters and inputs, the output of ONNX is inconstant with Tensorflow.
Urgency
none.
System information
-
OS Platform and Distribution (e.g., Linux Ubuntu 16.04) : (Red Hat 4.8.5-16)] on linux
-
ONNX Runtime installed from (source or binary): binary
-
ONNX Runtime version: onnx.version='1.6.0'
-
Python version: python3.65
-
Visual Studio version (if applicable):
-
GCC/Compiler version (if compiling from source): GCC 4.8.5 20150623
-
CUDA/cuDNN version:
torch.version: - 1.7.1+cu101
torch.version.cuda:10.2
torch.backends.cudnn.version():7603 -
GPU model and memory: Tesla V100, 64GB
To Reproduce
just single gru case.
Expected behavior
the output of onnx is equal with the output of tensorflow.
Screenshots

Additional context
GRU params:
Batch = 3
time step: T = 28
output size: H = 5
input size: X = 10

