Searched defs:conv_node (Results 1 – 9 of 9) sorted by relevance
38 int64_t GetOutChannels(const CNodePtr &conv_node) { in GetOutChannels()135 auto conv_node = pre_node->cast<CNodePtr>(); in Process() local180 bool ConvTransformFusion::AdjustActivationType(const CNodePtr &conv_node, const CNodePtr &transform… in AdjustActivationType() argument228 int ConvTransformFusion::GenNewConvTensor(const FuncGraphPtr &func_graph, const CNodePtr &conv_node… in GenNewConvTensor()322 int ConvTransformFusion::CalNewWeightTensor(const CNodePtr &conv_node, const tensor::TensorPtr &wei… in CalNewWeightTensor()
66 auto conv_node = pre_node->cast<CNodePtr>(); in Process() local
53 auto conv_node = tuple_cnode->input(1); in Process() local
72 auto conv_node = tuple_cnode->input(1); in Process() local
86 std::vector<int64_t> GenConvShape(const AnfNodePtr &conv_node) { in GenConvShape()100 nlohmann::json GenTuneInfo(const AnfNodePtr &conv_node, const std::map<AnfNodePtr, bool> &former_co… in GenTuneInfo()183 for (auto &conv_node : conv_list) { in TuneConvOps() local
93 auto conv_node = node->cast<CNodePtr>(); in Run() local
67 for (const auto &conv_node : conv_nodes_) { in CheckSplitValid() local107 int MultiConvSplit::GetMultiConvNodes(const AnfNodePtr &conv_node) { in GetMultiConvNodes()
385 bool CreateOutputsOfSplitWithOverlap(const FuncGraphPtr &func_graph, const AnfNodePtr &conv_node, in CreateOutputsOfSplitWithOverlap()
3125 void OnnxExporter::PrimConv2DTransposeExportHelper(const CNodePtr &conv_node, const CNodePtr &bias_… in PrimConv2DTransposeExportHelper()4237 auto conv_node = dyn_cast<CNode>(node->input(kOneNum)); in ExportMergeConv() local4345 auto conv_node = dyn_cast<CNode>(node->input(kOneNum)); in ExportMergeConv2DTranspose() local