Skip to content
Snippets Groups Projects

Draft: Feat/operator_unsqueeze

Closed Grégoire Kubler requested to merge (removed):feat/operator_squeeze_unsqueeze into dev
2 unresolved threads
Files
2
@@ -23,19 +23,17 @@
@@ -23,19 +23,17 @@
#include "aidge/utils/Registrar.hpp"
#include "aidge/utils/Registrar.hpp"
#include "aidge/utils/Types.h"
#include "aidge/utils/Types.h"
 
void Aidge::GlobalAveragePoolingImpl_cpu::forward() {
 
const GlobalAveragePooling_Op &op_ =
 
static_cast<const GlobalAveragePooling_Op &>(mOp);
 
// Check if input is provided
 
AIDGE_ASSERT(op_.getInput(0), "missing input 0");
void Aidge::GlobalAveragePoolingImpl_cpu::forward()
// Create the forward kernal with the wanted types
{
auto kernelFunc = Registrar<GlobalAveragePoolingImplForward_cpu>::create(
const GlobalAveragePooling_Op& op_ = static_cast<const GlobalAveragePooling_Op&>(mOp);
{op_.getInput(0)->dataType(), op_.getOutput(0)->dataType()});
// Check if input is provided
AIDGE_ASSERT(op_.getInput(0), "missing input 0");
// Create the forward kernal with the wanted types
// Call kernel
auto kernelFunc = Registrar<GlobalAveragePoolingImplForward_cpu>::create({op_.getInput(0)->dataType(),
kernelFunc(op_.getInput(0)->dims(), op_.getInput(0)->getImpl()->rawPtr(),
op_.getOutput(0)->dataType()});
op_.getOutput(0)->getImpl()->rawPtr());
}
Please register or sign in to reply
// Call kernel
kernelFunc(op_.getInput(0)->dims(),
op_.getInput(0)->getImpl()->rawPtr(),
op_.getOutput(0)->getImpl()->rawPtr());
}
\ No newline at end of file
Loading