Browse Source

Fix visit depend node

Signed-off-by: zhoufeng <zhoufeng54@huawei.com>
tags/v0.7.0-beta
zhoufeng 5 years ago
parent
commit
c48759f35e
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/backend/session/anf_runtime_algorithm.cc

+ 1
- 1
mindspore/ccsrc/backend/session/anf_runtime_algorithm.cc View File

@@ -405,7 +405,7 @@ KernelWithIndex AnfRuntimeAlgorithm::GetPrevNodeOutput(const AnfNodePtr &anf_nod
} }
auto node = cnode->input(input_idx + 1); auto node = cnode->input(input_idx + 1);
MS_EXCEPTION_IF_NULL(node); MS_EXCEPTION_IF_NULL(node);
return VisitKernel(node, 0);
return VisitKernelWithReturnType(node, 0);
} }


std::string AnfRuntimeAlgorithm::GetPrevNodeOutputFormat(const AnfNodePtr &anf_node, size_t input_idx) { std::string AnfRuntimeAlgorithm::GetPrevNodeOutputFormat(const AnfNodePtr &anf_node, size_t input_idx) {


Loading…
Cancel
Save