cpu_api.hpp

#include "cpu_op.hpp"



void cpu_api_conv2d_relu_maxpool2d(APICFG* pCFG);

void cpu_api_conv2d_relu(APICFG* pCFG);

void cpu_api_linear_relu(APICFG* pCFG);

void cpu_api_linear(APICFG* pCFG);

void cpu_api_logsoftmax(APICFG* pCFG);

void cpu_api_flatten(APICFG* api_cfg);