8000 concat op run on CPU · Issue #348 · airockchip/rknn-toolkit2 · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content
concat op run on CPU #348
New issue
New issue
Open
Open
concat op run on CPU#348
@zhaoxin111

Description

@zhaoxin111
zhaoxin111
opened on May 8, 2025

我通过torch感知量化训练了一个模型，转到rknn推理发现所有concat都跑在了CPU，而其余算子都运行在NPU.

我检查了下concat所有的输入输出量化参数都是统一的，实在看不出concat哪儿有问题。

我是用rknn-tool2.3转的rknn。

请问concat算子有什么需要特别注意的地方吗？

Image

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions
      0