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> |
||
|---|---|---|
| .. | ||
| CMakeLists.txt | ||
| DefaultDomainAtoF.cpp | ||
| DefaultDomainGtoP.cpp | ||
| DefaultDomainQtoZ.cpp | ||
| OnnxRecurrentLayerOpExpanders.cpp | ||
| PassDetail.h | ||
| Passes.cpp | ||
| Patterns.cpp | ||
| TorchOnnxToTorch.cpp | ||
| Utils.cpp | ||