diff --git a/include/tvm/runtime/packed_func.h b/include/tvm/runtime/packed_func.h index d2450c442048..47ada96a9b9f 100644 --- a/include/tvm/runtime/packed_func.h +++ b/include/tvm/runtime/packed_func.h @@ -233,7 +233,7 @@ class TypedPackedFunc { */ template >::value>::type> + std::function>::value>::type> TypedPackedFunc(const FLambda& typed_lambda) { // NOLINT(*) this->AssignTypedLambda(typed_lambda); } @@ -255,7 +255,7 @@ class TypedPackedFunc { */ template >::value>::type> + std::function>::value>::type> TSelf& operator=(FLambda typed_lambda) { // NOLINT(*) this->AssignTypedLambda(typed_lambda); return *this;