| /third_party/mindspore/mindspore/lite/tools/optimizer/parallel/ |
| D | conv2d_info.cc | 94 …auto conv_prim = GetValueNode<std::shared_ptr<ops::Conv2DFusion>>(cnode_->input(kAnfPrimitiveIndex… in CheckIfSplit()
102 auto input_node_abstract = GetCNodeInputAbstract(cnode_, 1); in CheckIfSplit()
104 auto weight_node_abstract = GetCNodeInputAbstract(cnode_, 2); in CheckIfSplit()
157 …auto conv_prim = GetValueNode<std::shared_ptr<ops::Conv2DFusion>>(cnode_->input(kAnfPrimitiveIndex… in CreateOutputsOfSplit()
209 if (cnode_ == nullptr) { in CheckConv2DPrimitiveType()
212 …if (!CheckPrimitiveType(cnode_, prim::kPrimConv2D) && !CheckPrimitiveType(cnode_, prim::kPrimConv2… in CheckConv2DPrimitiveType()
215 auto prim = GetValueNode<PrimitivePtr>(cnode_->input(kAnfPrimitiveIndex)); in CheckConv2DPrimitiveType()
246 …auto feature_split_cnode = CreateOutputsOfSplit(cnode_, 0, &feature_split_outputs, kAxisH, dev_num… in InferParallelCNodes()
262 …auto conv_prim = GetValueNode<std::shared_ptr<ops::Conv2DFusion>>(cnode_->input(kAnfPrimitiveIndex… in InferParallelCNodes()
276 std::string conv_cnode_name = cnode_->fullname_with_scope(); in ConstructOutputCNodes()
[all …]
|
| D | depthwise_conv2d_info.cc | 194 bool has_bias = cnode_->size() > kBiasIndex + 1; in CheckSplitOutputs()
276 std::string conv_cnode_name = cnode_->fullname_with_scope(); in AdJustInputs()
277 bool has_bias = cnode_->size() > kBiasIndex + 1; in AdJustInputs()
281 conv_inputs.push_back(cnode_->input(kWeightIndex + 1)); in AdJustInputs()
283 conv_inputs.push_back(cnode_->input(kBiasIndex + 1)); in AdJustInputs()
292 conv_inputs.push_back(cnode_->input(1)); in AdJustInputs()
335 …auto depth_wise_conv_prim = GetValueNode<std::shared_ptr<ops::Conv2DFusion>>(cnode_->input(kAnfPri… in CreateOutputsOfSplit()
397 auto prim = GetValueNode<PrimitivePtr>(cnode_->input(kAnfPrimitiveIndex)); in CheckDepthWiseConv2DPrimitiveType()
404 …if (!CheckPrimitiveType(cnode_, prim::kPrimConv2D) && !CheckPrimitiveType(cnode_, prim::kPrimConv2… in CheckDepthWiseConv2DPrimitiveType()
422 auto constant_node = cnode_->input(input_index + 1); in CreateConstantOutputsOfSplit()
[all …]
|
| D | operator_info.h | 54 cnode_(nullptr) {} in OperatorInfo()
87 CNodePtr cnode_{nullptr};
|
| D | operator_info.cc | 55 cnode_ = cnode; in Init()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ |
| D | akg_kernel_json_decoder.cc | 109 if (!DecodeInputDesc(cnode_json, func_graph) || cnode_ == nullptr) { in DecodeCNode()
119 return cnode_; in DecodeCNode()
194 cnode_ = func_graph->NewCNode(inputs); in DecodeInputDesc()
195 func_graph->AddNode(cnode_); in DecodeInputDesc()
210 nodes_map_[output_desc[kJsonKeyTensorName]] = cnode_; in DecodeOutputDesc()
219 …func_graph->NewCNode({NewValueNode(prim::kPrimTupleGetItem), cnode_, NewValueNode(SizeToLong(j))}); in DecodeOutputDesc()
232 const auto &inputs = cnode_->inputs(); in CreateKernelInfo()
247 if (AnfAlgo::GetCNodeName(cnode_) == prim::kPrimCast->name()) { in CreateKernelInfo()
248 AnfAlgo::SetNodeAttr(kIsBackendCast, MakeValue(false), cnode_); in CreateKernelInfo()
253 if (AnfAlgo::IsRealCNodeKernel(cnode_)) { in CreateKernelInfo()
[all …]
|
| D | akg_kernel_json_generator.cc | 49 cnode_ = anf_node->cast<CNodePtr>(); in Run()
50 MS_EXCEPTION_IF_NULL(cnode_); in Run()
52 op_info->set_op_name(AnfAlgo::GetCNodeName(cnode_)); in Run()
62 auto dyn_input_sizes = GetDynInputSize(cnode_); in ExtractInputs()
64 for (size_t i = 1; i < cnode_->size(); i++) { in ExtractInputs()
80 size_t output_tensor_num = AnfAlgo::GetOutputTensorNum(cnode_); in ExtractOutputs()
95 auto prim = GetCNodePrimitive(cnode_); in ExtractAttrs()
129 CNodePtr cnode_; member in mindspore::kernel::__anon7556c0a80111::OpInfoExtractor
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | onehot_info.cc | 133 if (!cnode_) { in ExtractInputInfo()
137 if (cnode_->inputs().size() != 5) { in ExtractInputInfo()
139 << cnode_->inputs().size(); in ExtractInputInfo()
150 …MS_LOG(WARNING) << "Input 2 of cnode is not a value node, its type is " << cnode_->input(2)->type_… in ExtractInputInfo()
215 Status status = ComputeReplaceGraph(cnode_); in Init()
|
| D | operator_info.h | 70 cnode_ = nullptr; in OperatorInfo()
159 void set_cnode(const CNodePtr &cnode) { cnode_ = cnode; } in set_cnode()
287 CNodePtr cnode_; variable
|
| D | dropout_do_mask_info.cc | 219 auto cnode = cnode_; in GetDropoutGenMaskReplaceOp()
296 auto cnode = cnode_; in ReplaceNodeInputOrAttrs()
|
| D | uniform_real_info.cc | 142 auto cnode = cnode_; in ReplaceNodeInputOrAttrs()
|
| D | batch_parallel_info.cc | 184 auto cnode = cnode_; in ReplaceNodeInputOrAttrs()
|
| D | tile_info.cc | 159 auto cnode = cnode_; in UpdateMultiples()
|
| D | operator_info.cc | 1725 MS_EXCEPTION_IF_NULL(cnode_); in cnode()
1726 return cnode_; in cnode()
|
| D | conv2d_info.cc | 1052 auto cnode = cnode_; in UpdateOutShape()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | generate_graph.cc | 136 cnode_ = cnode; in Init()
147 scope_ = cnode_->scope(); in Init()
154 instance_name_base_ = GetInstanceNameByCNode(cnode_); in Init()
|
| D | generate_graph.h | 59 CNodePtr cnode_;
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | manager.cc | 96 AddEdge(const CNodePtr &cnode, const AnfNodePtr &input) : cnode_{cnode}, input_{input} {} in AddEdge()
100 int index = static_cast<int>(cnode_->size()); in Apply()
101 cnode_->add_input(input_); in Apply()
103 counter->new_edges.add(std::move(cnode_), index, std::move(input_)); in Apply()
107 CNodePtr cnode_; member in mindspore::change::AddEdge
|