61 auto params = OpKernel::params_->fully();
65 const tensor_t&
W =
context.input(1);
66 const tensor_t* bias = params.has_bias_ ? &
context.input(2) :
nullptr;
74 const core::backend_t engine =
context.engine();
76 if (engine == core::backend_t::internal) {
77 kernels::fully_connected_op_internal(
80 params.has_bias_ ? (*bias)[0] : vec_t(),
85 else if (engine == core::backend_t::nnpack) {
86 kernels::fully_connected_op_nnpack(
89 params.has_bias_ ? (*bias)[0] : vec_t(),
94 else if (engine == core::backend_t::avx) {
95 kernels::fully_connected_op_avx(
98 params.has_bias_ ? (*bias)[0] : vec_t(),
104 throw nn_error(
"Not supported engine: " + to_string(engine));
Definition op_kernel.h:72