Bring Your Own Codegen to TVMDispatch Codegen Built Shared Library runtime::PackedFunc DNNLModule::GetFunction( const std::string& name, const std::shared_ptr& sptr_to_self) { if (name == "init") { return PackedFunc([sptr_to_self this->Init(args[0]); . }); } else { std::string curr_id = GetSubgraphID(name); return PackedFunc([sptr_to_self, curr_id, this](TVMArgs TVMRetValue* rv) { auto out = reinterpret_cast (args[args.size() - 1]>data); std::string encoded_name = kDnnlPrefix + curr_id; . auto func_s = reinter 0 码力 | 19 页 | 504.69 KB | 5 月前3
共 1 条
- 1













