Home
last modified time | relevance | path

Searched defs:pre_node (Results 1 – 23 of 23) sorted by relevance

/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Daffine_activation_fusion.cc55 AnfNodePtr pre_node = activation_node->input(1); in Process() local
Dconv_activation_fusion.cc54 AnfNodePtr pre_node = act_node->input(1); in Process() local
Daffine_fusion.cc64 AnfNodePtr pre_node = matmul_node->input(1); in Process() local
Dsqueeze_fusion.cc84 auto pre_node = squeeze_cnode->input(1); in Process() local
Dtranspose_fusion.cc158 auto pre_node = trans_cnode_2->input(1); in TransTransFusion() local
Dconv_transform_fusion.cc103 auto pre_node = transform_node->input(1); in Process() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.cc197 void InsertNode(const Operator &op, const CNodePtr &node, size_t index, const AnfNodePtr &pre_node, in InsertNode()
231 static CNodePtr ReplaceNode(const Operator &op, const AnfNodePtr &pre_node, const FuncGraphPtr &fun… in ReplaceNode()
328 const FuncGraphPtr &func_graph, int64_t pos, const CNodePtr &pre_node) { in InsertRedistribution()
385 AnfNodePtr pre_node = node->input(LongToSize(pos)); in InsertGetTensorSliceOp() local
431 const CNodePtr &pre_node) { in Redistribution()
544 … const TensorRedistribution &tensor_redistribution, const CNodePtr &pre_node) { in StepRedistribution()
794 auto pre_node = node->input(LongToSize(replace_input.second)); in StepReplaceGraph() local
891 auto pre_node = cnode->input(last_indexs[last_node_index]); in InsertVirtualOutput() local
1039 auto pre_node = node->input(index); in InsertMirrorBeforeCast() local
1172 AnfNodePtr pre_node = node->input(index); in InsertMirrorOps() local
[all …]
Dstep_auto_parallel.cc860 auto pre_node = cnode->input(1); in ReshapeCostCompute() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_build_ascend.cc131 static std::vector<size_t> CalCleanZerosSize(const CNodePtr &pre_node) { in CalCleanZerosSize()
160const mindspore::CNodePtr &pre_node, std::vector<mindspore::CNodePtr> *new_nodes) { in AddTbeClearZeroNode()
Dascend_stream_assign.cc1044 auto pre_node = AnfAlgo::GetCNodeName(cnode_ptr_list[i - 1]); in InsertStreamActiveForCommon() local
/third_party/musl/porting/linux/user/ldso/
Ddynlink_rand.c80 struct handle_node *pre_node = NULL; in remove_handle_node() local
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dredundant_op_remove_pass.cc78 AnfNodePtr pre_node = cnode->input(1); in ProcessDependencyWithTwoNodes() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dnode_info.cc333 auto pre_node = cnode->input(1); in FindReshapePreNodeStraCosts() local
Dpipeline_split_utils.cc57 auto pre_node = cnode->input(1); in FindAccuGrad() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/tasksink/
Dtask_generator.cc108 auto pre_node = anf_node_ptr->input(i + 1)->cast<CNodePtr>(); in LaunchAddrCleanKernel() local
/third_party/mindspore/mindspore/lite/tools/converter/acl/
Dacl_pass.cc266 AnfNodePtr pre_node = nullptr; in TraceOutput() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_runtime.cc1189 auto pre_node = (cnode->inputs()[1])->cast<CNodePtr>(); in GenAddrCleanLaunchArgs() local
1627 auto pre_node = AnfAlgo::GetPrevNodeOutput(node, 1, true); in CheckSparsePSEmbeddingCache() local
Dkernel_adjust.cc185 auto pre_node = orders[idx - 1]; in CopyMemcpyList() local
/third_party/mindspore/tests/ut/cpp/parallel/
Dstep_parallel_test.cc430 CNodePtr pre_node = inputs[1]->cast<CNodePtr>(); in TEST_F() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dparallel_fusion.cc62 for (const auto &pre_node : predecessors) { in ProcessThroughPassCNode() local
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dconvert.cc625 AnfNodePtr pre_node = nullptr; in TraceOutput() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/somas/
Dsomas.cc589 auto pre_node = kernel->input(i + 1)->cast<CNodePtr>(); in InitAtomicCleanInputs() local
/third_party/mindspore/mindspore/ccsrc/backend/session/
Danf_runtime_algorithm.cc2444 auto pre_node = (cnode->inputs()[1])->cast<CNodePtr>(); in CacheAddrForAtomicClean() local