| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/ |
| D | tbe_kernel_select.cc | 55 : cnode_ptr_(std::move(kernel_node)), kernel_info_list_(kernel_info_list) {} in TbeKernelSelect()
58 MS_EXCEPTION_IF_NULL(cnode_ptr_); in TbeMetadataInfoEx()
60 node_name_ = AnfAlgo::GetCNodeName(cnode_ptr_); in TbeMetadataInfoEx()
61 full_name_ = cnode_ptr_->fullname_with_scope(); in TbeMetadataInfoEx()
63 auto op_info_ptr = tbe::TbeDynamicShapeUtil::FindOp(node_name_, cnode_ptr_); in TbeMetadataInfoEx()
67 if (!TbePropertyChecker::CheckTbeProperties(cnode_ptr_)) { in TbeMetadataInfoEx()
95 size_t real_input_tensor_num = AnfAlgo::GetInputTensorNum(cnode_ptr_); in GetCommonPatternKernelInfo()
97 size_t real_output_tensor_num = AnfAlgo::GetOutputTensorNum(cnode_ptr_); in GetCommonPatternKernelInfo()
100 …MS_LOG(EXCEPTION) << AnfAlgo::GetCNodeName(cnode_ptr_) << "'s op info input & output is null, plea… in GetCommonPatternKernelInfo()
147 auto format = AnfAlgo::GetPrevNodeOutputFormat(cnode_ptr_, 0); in GetAgnosticPatternKernelInfo()
[all …]
|
| D | tbe_kernel_reduce_selecter.cc | 38 auto input_num = AnfAlgo::GetInputTensorNum(cnode_ptr_); in GetShapeInfo()
39 auto output_num = AnfAlgo::GetOutputTensorNum(cnode_ptr_); in GetShapeInfo()
45 input_shape_ = AnfAlgo::GetPrevNodeOutputInferShape(cnode_ptr_, kInputIndex_0); in GetShapeInfo()
47 output_shape_ = AnfAlgo::GetOutputInferShape(cnode_ptr_, kOutputIndex_0); in GetShapeInfo()
52 axis_ = GetReduceAttrAxis(cnode_ptr_); in GetShapeInfo()
136 if (!AnfAlgo::HasNodeAttr(kAttrKeepDims, cnode_ptr_)) { in GetReduceAttrKeepDim()
141 keep_dims_ = AnfAlgo::GetNodeAttr<bool>(cnode_ptr_, kAttrKeepDims); in GetReduceAttrKeepDim()
|
| D | tbe_kernel_broadcast_selecter.cc | 36 if (AnfAlgo::HasNodeAttr(kAttrDynInputSizes, cnode_ptr_)) { in GetShapeInfo()
37 …auto dynamic_size_vec = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(cnode_ptr_, kAttrDynInputSizes); in GetShapeInfo()
40 …MS_LOG(EXCEPTION) << "Node [" << AnfAlgo::GetCNodeName(cnode_ptr_) << "]'s attr [dyn_input_sizes] … in GetShapeInfo()
43 MS_LOG(EXCEPTION) << "Node [" << AnfAlgo::GetCNodeName(cnode_ptr_) in GetShapeInfo()
46 auto dynamic_input_shape0_ = AnfAlgo::GetPrevNodeOutputInferShape(cnode_ptr_, kInputIndex_0); in GetShapeInfo()
51 input_num_ = AnfAlgo::GetInputTensorNum(cnode_ptr_); in GetShapeInfo()
53 auto input_shape = AnfAlgo::GetPrevNodeOutputInferShape(cnode_ptr_, i); in GetShapeInfo()
59 output_num_ = AnfAlgo::GetOutputTensorNum(cnode_ptr_); in GetShapeInfo()
61 auto output = AnfAlgo::GetOutputInferShape(cnode_ptr_, i); in GetShapeInfo()
|
| D | tbe_kernel_reduce_selecter.h | 28 explicit TbeKernelReduceSelecter(CNodePtr cnode_ptr) : cnode_ptr_(std::move(cnode_ptr)) {} in TbeKernelReduceSelecter()
44 CNodePtr cnode_ptr_; variable
|
| D | tbe_kernel_broadcast_selecter.h | 30 explicit TbeKernelBroadCastSelecter(CNodePtr cnode_ptr) : cnode_ptr_(std::move(cnode_ptr)) {} in TbeKernelBroadCastSelecter()
49 CNodePtr cnode_ptr_; variable
|
| D | tbe_kernel_select.h | 71 CNodePtr cnode_ptr_;
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/ |
| D | ai_cpu_dynamic_kernel.cc | 49 auto cnode = cnode_ptr_.lock(); in Execute()
62 auto cnode = cnode_ptr_.lock(); in Initialize()
113 auto cnode = cnode_ptr_.lock(); in UpdateInputOutputAddr()
147 auto cnode = cnode_ptr_.lock(); in UpdateExtInfo()
184 auto cnode = cnode_ptr_.lock(); in UpdateOutputShapeFromExtInfo()
210 AnfAlgo::SetOutputInferTypeAndShape(type_ids, shapes, cnode_ptr_.lock().get()); in UpdateOutputShapeFromExtInfo()
215 auto cnode = cnode_ptr_.lock(); in PostExecute()
|
| D | ai_core_dynamic_kernel.cc | 46 auto cnode = cnode_ptr_.lock(); in Execute()
69 auto cnode = cnode_ptr_.lock(); in ParseCompileJson()
100 auto cnode = cnode_ptr_.lock(); in UpdateArgs()
133 auto cnode = cnode_ptr_.lock(); in ComputeTiling()
150 auto cnode = cnode_ptr_.lock(); in AllocateWorkspace()
|
| D | hccl_dynamic_kernel.cc | 39 auto cnode = cnode_ptr_.lock(); in UpdateArgs()
79 auto cnode = cnode_ptr_.lock(); in StaticShapeExecute()
91 auto cnode = cnode_ptr_.lock(); in Execute()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/executor/ |
| D | dynamic_kernel.cc | 33 auto cnode = cnode_ptr_.lock(); in Initialize()
55 int DynamicKernel::GetKernelType() const { return AnfAlgo::GetKernelType(cnode_ptr_.lock()); } in GetKernelType()
59 auto cnode = cnode_ptr_.lock(); in RebuildDependTensor()
82 auto cnode = cnode_ptr_.lock(); in InferShape()
96 auto cnode = cnode_ptr_.lock(); in InferShapeRecursive()
|
| D | dynamic_kernel.h | 34 cnode_ptr_(cnode_ptr), in DynamicKernel()
54 const CNodeWeakPtr cnode_ptr_; variable
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/allreduce_fusion/ |
| D | allreduce_node.h | 35 …: cnode_ptr_(nullptr), prev_(), next_(), paras_(), para_size_map_(), curr_para_size_(0), depend_fe… in AllreduceNode()
55 CNodePtr cnode_ptr_;
|
| D | allreduce_node.cc | 71 cnode_ptr_ = cnode_ptr; in Init()
117 MS_LOG(INFO) << "cnode: " << cnode_ptr_->DebugString() << "para size: " << paras_.size(); in ToString()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/ |
| D | gpu_kernel.cc | 26 auto cnode = cnode_ptr_.lock(); in UpdateArgs()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/ |
| D | dynamic_shape_kernel.cc | 24 auto cnode = cnode_ptr_.lock(); in Execute()
|
| D | dynamic_broadcast_gradient_args_kernel.cc | 191 auto cnode = cnode_ptr_.lock(); in Execute()
|