mirror of https://github.com/llvm/torch-mlir
4a0bed0ce0
This commit extends the OnnxToTorch lowering for BatchNormalization op for supporting the case when training=True. Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com> |
||
---|---|---|
.. | ||
ops | ||
simple_ops_a_to_f.mlir | ||
simple_ops_g_to_p.mlir | ||
simple_ops_q_to_z.mlir | ||
unsupported_fb_opt_ops.mlir |