Searched refs:cnode_refcount (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | ascend_session.h | 77 const std::map<KernelWithIndex, size_t> &cnode_refcount) override;
|
| D | ascend_session.cc | 179 const std::map<KernelWithIndex, size_t> &cnode_refcount, in GenOpOutputStubTensor() argument
188 if (cnode_refcount.find(kernel_with_index) == cnode_refcount.end()) { in GenOpOutputStubTensor()
1119 const std::map<KernelWithIndex, size_t> &cnode_refcount) { in BuildOpsInGraph() argument
1144 GenOpOutputStubTensor(single_op_graph_iter->second, kernel, cnode_refcount, &op_output_info); in BuildOpsInGraph()
1150 GenOpOutputStubTensor(single_op_graph, kernel, cnode_refcount, &op_output_info); in BuildOpsInGraph()
|
| D | session_basic.h | 244 const std::map<KernelWithIndex, size_t> &cnode_refcount) {} in BuildOpsInGraph() argument
|
| D | session_basic.cc | 2281 std::map<KernelWithIndex, size_t> cnode_refcount; in RunOpsInGraphImpl() local
2282 GetRefCount(kernel_graph.get(), &cnode_refcount); in RunOpsInGraphImpl()
2283 BuildOpsInGraph(graph_id, parameter_index, inputs, cnode_refcount); in RunOpsInGraphImpl()
2307 HandleOpInputs(input_tensor_info.input_kernel, &cnode_refcount, &op_output_map); in RunOpsInGraphImpl()
2308 HandleOpOutputs(kernel, op_outputs, cnode_refcount, &op_output_map, &graph_output_info); in RunOpsInGraphImpl()
|