47 #include "tiny_dnn/core/framework/op_kernel.h"
49 #include "tiny_dnn/core/kernels/maxpool_op_internal.h"
50 #include "tiny_dnn/core/kernels/maxpool_op_nnpack.h"
51 #include "tiny_dnn/core/kernels/maxpool_op_avx.h"
61 auto& params = OpKernel::params_->maxpool();
64 const tensor_t& in_data = context.input(0);
65 tensor_t& out_data = context.output(1);
68 fill_tensor(out_data, float_t(0));
73 const core::backend_t engine = context.engine();
75 if (engine == core::backend_t::internal) {
76 kernels::maxpool_op_internal(
81 context.parallelize());
82 }
else if (engine == core::backend_t::nnpack) {
95 kernels::maxpool_op_nnpack(
99 }
else if (engine == core::backend_t::avx) {
100 kernels::maxpool_op_avx(
105 context.parallelize());
107 throw nn_error(
"Not supported engine: " + to_string(engine));
Definition: maxpool_op.h:55
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