Skip to content

Learning backend cuda

Houssem ROUIS requested to merge hrouis/aidge_onnx:learning_backend_cuda into dev

Context

Add import/export for following operators:

  • And
  • ArgMax
  • ReduceSum

and add noop_with_empty_axes attr to reducemean operator

TODO

  • And
  • ArgMax
  • ReduceMean
  • ReduceSum

Merge request reports

Loading