implement conv1D operator cpp export and unit tests for it
2 unresolved threads
2 unresolved threads
Context
implement conv1D operator cpp export and unit tests for it.
Needed aidge_core modification to "support dim==3 tensors for node export (format NCW)" (see MR : aidge_core!446 (closed))
Modified files
(Summarize the changes you made for each modified file - example:
-
aidge_export_cpp/kernels/conv1D.hppkernel implementation inspired fromconvolution.hpp -
aidge_export_cpp/templates/configuration/conv1D_config.jinjajinja config inspired fromconvolution_config.jinja -
aidge_export_cpp/templates/kernel_forward/conv1D_forward.jinjajinja forward inspired fromconvolution_forward.jinja -
aidge_export_cpp/operators/Conv1D.pyinspired from Conv.py -
aidge_export_cpp/unit_tests/test_export.pyadded 3 tests for conv1d export, with stride and dilation.
The 3 tests results OK
Merge request reports
Activity
Filter activity
mentioned in merge request aidge_core!446 (closed)
assigned to @louislerbourg
requested review from @olivierbichler
added Feature 🚀 label
478 478 479 479 self.unit_test_export(model, "Conv2D", [[1, 3, 12, 12]], False, False) 480 480 481 def test_conv1D(self): 482 print("Conv1D") 483 model = aidge_core.sequential([ 484 aidge_core.Conv1D(in_channels=3, out_channels=3, kernel_dims=[3],name="conv") 485 ]) 486 487 self.unit_test_export(model, "Conv1D", [[1, 3, 12]], False, False) - aidge_export_cpp/operators/Conv1D.py 0 → 100644
1 import aidge_core 2 from aidge_core.export_utils import ExportNodeCpp, get_node_from_metaop 3 from aidge_export_cpp import ROOT, ExportLibCpp, set_scaling_attributes 4 5 @ExportLibCpp.register("Conv1D", aidge_core.ImplSpec(aidge_core.IOSpec(aidge_core.dtype.any))) added StatusChanges required label
mentioned in merge request !53 (merged)
Please register or sign in to reply