The GitHub Actions job "Teams" on tvm.git/main has succeeded. Run started by GitHub user ysh329 (triggered by ysh329).
Head commit for run: 4ab312bf12cbe1f1da87c7a9687287c938503aab / Wei-Cheng Hsu <[email protected]> [Relax][Frontend][TFLite] Add Conv3D support (#19523) Description This PR adds support for the CONV_3D operator in the TFLite frontend for Relax. Key Changes - Operator Mapping: Added CONV_3D to the OperatorConverter mapping in tflite_frontend.py. - Implementation: - Implemented convert_conv3d to handle 3D convolution attributes such as StrideD/H/W, DilationD/H/W, and Padding. - Correctly handled the TFLite 3D kernel layout, which is expected to be DHWIO (Depth, Height, Width, Input Channels, Output Channels). - Integrated support for fused activation functions (ReLU, ReLU6, etc.) directly following the convolution. - Unit Tests: - Added comprehensive tests in tests/python/relax/test_frontend_tflite.py covering: - VALID and SAME padding modes. - Various stride and dilation configurations. - Verification against expected Relax IR structure. Testing: - `python3 -m pytest tests/python/relax/test_frontend_tflite.py -k "test_conv3d"` Notes for Reviewers The implementation follows the existing pattern used for CONV_2D but extends it to the 5D case (NDHWC layout). I've ensured that the kernel layout mapping aligns with TVM's R.nn.conv3d requirements. Related to: https://github.com/apache/tvm/issues/19519 Report URL: https://github.com/apache/tvm/actions/runs/25604496794 With regards, GitHub Actions via GitBox --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
