Searched refs:replace_op_ (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/lite/tools/optimizer/parallel/ |
| D | operator_info.h | 52 replace_op_(nullptr), in OperatorInfo()
60 AnfNodePtr replace_op() const { return replace_op_; } in replace_op()
84 AnfNodePtr replace_op_{nullptr};
|
| D | conv2d_info.cc | 371 replace_op_ = CreateReduceNode(cnode_, parallel_output_nodes_, dev_num); in InferReplaceOp()
382 replace_op_ = CreateConcateNode(cnode_, parallel_output_nodes_, concat_dim, dev_num); in InferReplaceOp()
385 if (replace_op_ == nullptr) { in InferReplaceOp()
|
| D | depthwise_conv2d_info.cc | 499 replace_op_ = CreateConcateNode(cnode_, parallel_output_nodes_, split_dim_, dev_num); in InferReplaceOp()
500 if (replace_op_ == nullptr) { in InferReplaceOp()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | batchnorm_info.cc | 200 replace_op_.clear(); in InferReplaceOps()
234 replace_op_ = {std::make_pair(SYNC_BATCH_NORM, args)}; in InferReplaceOps()
|
| D | reshape_info.cc | 131 replace_op_ = constructor.SkipRedisReshapeOP(output_layout_.slice_shape().array()); in ComputeReplaceOp()
145 replace_op_ = redistribution_oplist_ptr->first; in ComputeReplaceOp()
148 MS_LOG(DEBUG) << name_ << ": replace op size = " << replace_op_.size(); in ComputeReplaceOp()
|
| D | operator_info.h | 115 ForwardOp replace_op() const { return replace_op_; } in replace_op()
246 ForwardOp replace_op_; variable
|
| D | activation_info.cc | 321 replace_op_ = {std::make_pair(DROPOUT, args)}; in InferReplaceOps()
546 replace_op_ = {std::make_pair(SQUEEZE, args)}; in InferReplaceOps()
|
| D | get_next_info.cc | 217 replace_op_ = {std::make_pair(GET_NEXT, args)}; in InferReplaceOps()
|
| D | gather_v2_p_info.cc | 849 replace_op_.push_back(op); in ComputeReplaceOp()
|
| D | operator_info.cc | 145 replace_op_.clear(); in ResetQueueMember()
|