47 #include "tiny_dnn/core/framework/op_kernel.h"
49 #include "tiny_dnn/core/kernels/maxpool_op_avx.h"
50 #include "tiny_dnn/core/kernels/maxpool_op_internal.h"
60 auto& params = OpKernel::params_->maxpool();
63 tensor_t& prev_delta = context.input_grad(0);
64 tensor_t& curr_delta = context.output_grad(1);
67 fill_tensor(prev_delta, float_t(0));
71 const core::backend_t engine = context.engine();
73 if (engine == core::backend_t::internal) {
74 kernels::maxpool_grad_op_internal(
79 context.parallelize());
80 }
else if (engine == core::backend_t::avx) {
81 kernels::maxpool_grad_op_avx(
86 context.parallelize());
88 throw nn_error(
"Not supported engine: " + to_string(engine));
Definition: maxpool_grad_op.h:54
Definition: op_kernel.h:55
Definition: op_kernel.h:72
Definition: op_kernel.h:175
error exception class for tiny-dnn
Definition: nn_error.h:37