mirror of https://github.com/llvm/torch-mlir
a1fe307a76
If there is only a single value scattered there can be an implicit batch dimension. This includes a check for the implicit batch dimension when reshaping the update tensor. It includes an e2e test to verify correctness. |
||
---|---|---|
.. | ||
CMakeLists.txt | ||
TorchToTMTensor.cpp |