feat: add AveragePooling2D layer support to backend#267
feat: add AveragePooling2D layer support to backend#267paramananda-15 wants to merge 1 commit intoc2siorg:mainfrom
Conversation
|
The backend model generation code for |
78864bc to
7c8c95d
Compare
feat: add AveragePooling2D frontend support fix: sync package-lock.json with dependencies fix: regenerate package-lock.json to match dependencies
a45a081 to
789f7ca
Compare
Added full stack support for AveragePooling2D:
Also fixed CI issues (lock file + formatting) and squashed commits. Ready for review. |
Description
Adds support for the AveragePooling2D layer in the TensorMap backend model generation service.
This change allows ReactFlow graphs to include AveragePooling nodes that are translated into
tf.keras.layers.AveragePooling2Dwhen constructing the Keras functional model.The implementation adds handling for a new node type (
customavgpool) inapp/services/model_generation.py, enabling TensorMap to support average pooling operationscommonly used in convolutional neural networks.
This contribution aligns with the TensorMap GSoC project goal of expanding the
supported layer types in the backend.
Fixes #N/A
Type of Change
How Has This Been Tested?
All existing backend tests were executed locally after implementing the change
to ensure no regressions were introduced.
Command used:
python -m pytest
Results:
130 passed in ~3 seconds
Screenshots (if applicable)
N/A
Checklist