This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
AI流水线
Repositories
Datasets
Forum
实训
竞赛
大数据
AI开发
Register
Sign In
Huawei_Technology
/
MindSpore-GraphEngine
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
10
Wiki
evaluate
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
HPC
Browse Source
Feature: reset shape of dynamic single op
pull/618/head
l00444296
5 years ago
parent
d71712b695
commit
71fff26293
1 changed files
with
2 additions
and
2 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+2
-2
ge/graph/passes/dynamic_single_op_reset_shape_pass.cc
+ 2
- 2
ge/graph/passes/dynamic_single_op_reset_shape_pass.cc
View File
@@ -26,8 +26,8 @@
namespace ge {
namespace {
const int64_t kDynamicShapeDim = -2;
const char *const kEngineNameAiCpu = "
aicpu_ascend_kernel
";
const char *const kEngineNameAiCpuTf = "
aicpu_tf_kernel
";
const char *const kEngineNameAiCpu = "
DNN_VM_AICPU_ASCEND
";
const char *const kEngineNameAiCpuTf = "
DNN_VM_AICPU
";
} // namespace
Status DynamicSingleOpResetShapePass::Run(ComputeGraphPtr graph) {
GE_CHECK_NOTNULL(graph);
Write
Preview
Loading…
Cancel
Save