torch-mlir/test/Conversion/TorchOnnxToTorch
Vivek Khandelwal 4a0bed0ce0
[ONNX] Add training mode support for BatchNormalization op (#3597)
This commit extends the OnnxToTorch lowering for BatchNormalization op
for supporting the case when training=True.

Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>
2024-08-14 10:46:38 +05:30
..
ops Support onnx.If (#2825) 2024-04-30 18:36:40 +00:00
simple_ops_a_to_f.mlir [ONNX] Add training mode support for BatchNormalization op (#3597) 2024-08-14 10:46:38 +05:30
simple_ops_g_to_p.mlir [onnx] Support integer types for `onnx.Pow` (#3626) 2024-08-13 09:39:04 -07:00
simple_ops_q_to_z.mlir [ONNX] add support for tfidfvectorizer (#3553) 2024-08-12 18:10:11 -05:00
unsupported_fb_opt_ops.mlir Add attributes support for onnx cumsum op (#3241) 2024-05-11 02:09:01 +08:00