Skip to content
Snippets Groups Projects

Draft: [Issue #251] : WIP: Softmax Backward implementation for cpu

Open Adam Maroni requested to merge adamaroni/aidge_backend_cpu:feat_operator_SoftMax into dev
3 unresolved threads
4 files
+ 295
13
Compare changes
  • Side-by-side
  • Inline
Files
4
@@ -23,7 +23,8 @@
namespace Aidge {
// Operator implementation entry point for the backend
using SoftmaxImpl_cpu = OperatorImpl_cpu<Softmax_Op,
void(std::size_t, const std::vector<DimSize_t>&, const void*, void*)>;
void(std::size_t, const std::vector<DimSize_t>&, const void*, void*),
void(std::size_t axisIdx, const std::vector<DimSize_t>& , const void* , const void* , void*)>;
// Implementation entry point registration to Operator
REGISTRAR(Softmax_Op, "cpu", Aidge::SoftmaxImpl_cpu::create);
Loading