Searched defs:conv_node (Results 1 – 8 of 8) sorted by relevance
32 int64_t GetOutChannels(const CNodePtr &conv_node) { in GetOutChannels()104 auto conv_node = pre_node->cast<CNodePtr>(); in Process() local173 int ConvTransformFusion::GenNewConvTensor(const FuncGraphPtr &func_graph, const CNodePtr &conv_node… in GenNewConvTensor()267 int ConvTransformFusion::CalNewWeightTensor(const CNodePtr &conv_node, const tensor::TensorPtr &wei… in CalNewWeightTensor()
59 auto conv_node = pre_node->cast<CNodePtr>(); in Process() local
51 auto conv_node = tuple_cnode->input(1); in Process() local
66 auto conv_node = tuple_cnode->input(1); in Process() local
89 auto conv_node = node->cast<CNodePtr>(); in Run() local
65 for (const auto &conv_node : conv_nodes_) { in CheckSplitValid() local105 int MultiConvSplit::GetMultiConvNodes(const AnfNodePtr &conv_node) { in GetMultiConvNodes()
358 bool CreateOutputsOfSplitWithOverlap(const FuncGraphPtr &func_graph, const AnfNodePtr &conv_node, in CreateOutputsOfSplitWithOverlap()
1701 auto conv_node = dyn_cast<CNode>(node->input(kOneNum)); in ExportMergeConv() local