Browse Source

Feature: reset shape of dynamic single op

pull/618/head
l00444296 5 years ago
parent
commit
71fff26293
1 changed files with 2 additions and 2 deletions
  1. +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);


Loading…
Cancel
Save