Home
last modified time | relevance | path

Searched refs:AnfNode (Results 1 – 25 of 39) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_build.h58 static bool GenFusionDataInputJson(const std::shared_ptr<mindspore::AnfNode> &data_input,
74 static void GenDescJson(const std::shared_ptr<mindspore::AnfNode> &anf_node, size_t node_out_idx,
77 …static void GenFusionOutputDescJson(const std::shared_ptr<mindspore::AnfNode> &anf_node, size_t no…
81 static void GenReusedOutputDesc(const std::shared_ptr<mindspore::AnfNode> &anf_node, size_t index,
100 …bool GenTbeSingleKernelJson(const std::shared_ptr<AnfNode> &anf_node, nlohmann::json *kernel_json);
102 …void GenTbeAttrJson(const std::shared_ptr<AnfNode> &anf_node, const std::shared_ptr<OpInfo> &op_in…
107 …bool GenTbeInputsJson(const std::shared_ptr<AnfNode> &anf_node, const std::shared_ptr<OpInfo> &op_…
109 …bool GenTbeOutputsJson(const std::shared_ptr<AnfNode> &anf_node, const std::shared_ptr<OpInfo> &op…
114 …bool GenInputDescJson(const std::shared_ptr<AnfNode> &anf_node, size_t real_input_index, bool valu…
117 bool GenOutputDescJson(const std::shared_ptr<AnfNode> &anf_node,
[all …]
Dtbe_adapter.h57 …static void InputOrderPass(const std::shared_ptr<AnfNode> &anf_node, std::vector<T> const &inputs_… in InputOrderPass()
97 …static bool DynamicInputAdjusted(const std::shared_ptr<AnfNode> &anf_node, std::vector<T> const &i… in DynamicInputAdjusted()
122 static void FusionInputOrderPass(const std::shared_ptr<AnfNode> &anf_node,
125 static void InputOrderPass(const std::shared_ptr<AnfNode> &anf_node,
127 static bool DynamicInputAdjusted(const std::shared_ptr<AnfNode> &anf_node,
132 …static void GenTopKV2IndicesTensorInfo(const std::shared_ptr<AnfNode> &anf_node, size_t real_input…
Dtbe_kernel_build.cc188 bool TbeKernelJsonCreator::GenTbeSingleKernelJson(const std::shared_ptr<mindspore::AnfNode> &anf_no… in GenTbeSingleKernelJson()
284 void TbeKernelJsonCreator::GenValidInputDescJson(const std::shared_ptr<AnfNode> &anf_node, size_t r… in GenValidInputDescJson()
316 bool TbeKernelJsonCreator::GenInputDescJson(const std::shared_ptr<AnfNode> &anf_node, size_t real_i… in GenInputDescJson()
342 bool TbeKernelJsonCreator::GenInputList(const std::shared_ptr<AnfNode> &anf_node, size_t input_tens… in GenInputList()
390 bool GetInputNameAndRealNum(const std::shared_ptr<AnfNode> &anf_node, const std::shared_ptr<OpIOInf… in GetInputNameAndRealNum()
420 bool TbeKernelJsonCreator::GenTbeInputsJson(const std::shared_ptr<AnfNode> &anf_node, in GenTbeInputsJson()
457 bool TbeKernelJsonCreator::GenTbeOutputsJson(const std::shared_ptr<AnfNode> &anf_node, in GenTbeOutputsJson()
471 const std::shared_ptr<mindspore::AnfNode> &anf_node, in GenOutputDescJson()
509 void TbeKernelJsonCreator::GenOutputList(const std::shared_ptr<AnfNode> &anf_node, const size_t &ou… in GenOutputList()
544 void TbeKernelJsonCreator::GenTbeAttrJson(const std::shared_ptr<AnfNode> &anf_node, in GenTbeAttrJson()
[all …]
Dtbe_adapter.cc79 bool TbeAdapter::DynamicInputAdjusted(const std::shared_ptr<AnfNode> &anf_node, in DynamicInputAdjusted()
103 void TbeAdapter::InputOrderPass(const std::shared_ptr<AnfNode> &anf_node, in InputOrderPass()
143 void TbeAdapter::FusionInputOrderPass(const std::shared_ptr<AnfNode> &anf_node, in FusionInputOrderPass()
255 void TbeAdapter::GenTopKV2IndicesTensorInfo(const std::shared_ptr<mindspore::AnfNode> &anf_node, in GenTopKV2IndicesTensorInfo()
Dtbe_kernel_parallel_build.h57 const std::vector<size_t> &output_size_list, AnfNode *node) const;
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Ddevice_tensor_store.h44 void Insert(AnfNode *key, const DeviceTensorPtr &value) { in Insert()
61 void Remove(AnfNode *key) { in Remove()
69 std::vector<DeviceTensorPtr> Fetch(AnfNode *key) const { in Fetch()
80 DeviceTensor *Fetch(AnfNode *key, DeviceTensorType value_type) const { in Fetch()
103 std::unordered_map<AnfNode *, std::vector<DeviceTensorPtr>> device_tensors_;
/third_party/mindspore/mindspore/core/ir/
Danf_extends.cc31 TypePtr AnfNode::Type() const { return (abstract_ == nullptr) ? nullptr : abstract_->BuildType(); } in Type()
32 BaseShapePtr AnfNode::Shape() const { return (abstract_ == nullptr) ? nullptr : abstract_->BuildSha… in Shape()
34 std::string AnfNode::ToString() const { in ToString()
35 …return mindspore::label_manage::Label(const_cast<AnfNode *>(this)->shared_from_base<AnfNode>()->de… in ToString()
Danf.h99 class MS_CORE_API AnfNode : public Base {
101 explicit AnfNode(const FuncGraphPtr &func_graph) in AnfNode() function
107 hash_(std::hash<const AnfNode *>()), in AnfNode()
116 ~AnfNode() override = default;
117 MS_DECLARE_PARENT(AnfNode, Base);
141 debug_info_->set_node(shared_from_base<AnfNode>()); in debug_info()
149 debug_info_->set_node(shared_from_base<AnfNode>()); in set_debug_info()
166 virtual bool operator==(const AnfNode &other) const { return &other == this; }
167 friend std::ostream &operator<<(std::ostream &os, const AnfNode &node) {
225 std::hash<const AnfNode *> hash_;
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/session/
Danf_runtime_algorithm.h200 static void SetOutputAddr(const DeviceAddressPtr &addr, size_t output_idx, AnfNode *node);
202 static void SetWorkspaceAddr(const DeviceAddressPtr &addr, size_t output_idx, AnfNode *node);
209 … const std::vector<std::vector<size_t>> &shapes, AnfNode *node);
214 … const std::vector<abstract::BaseShapePtr> &shapes, AnfNode *node);
215 static void CopyAbstract(const AnfNodePtr &from_node, AnfNode *to_node);
228 …etSelectKernelBuildInfo(const kernel::KernelBuildInfoPtr &select_kernel_build_info, AnfNode *node);
234 static void SetKernelMod(const kernel::KernelModPtr &kernel_mod, AnfNode *node);
250 static void SetStreamId(uint32_t stream_id, AnfNode *node);
254 static void SetStreamDistinctionLabel(uint32_t stream_label, AnfNode *node);
256 static uint32_t GetStreamDistinctionLabel(const AnfNode *node);
[all …]
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dconvert.h143 std::unordered_map<AnfNode *, std::string> op_draw_name_;
150 …OutHandler GetHandler(const AnfNodePtr &node, const std::stack<uint64_t> &index_stack, AnfNode *co…
199 std::unordered_map<AnfNode *, DfGraph> branches_map_;
200 std::unordered_map<AnfNode *, OperatorPtr> op_cache_;
201 std::unordered_map<AnfNode *, std::vector<ControlEdge>> control_edge_cache_;
204 std::unordered_map<AnfNode *, OutHandler> out_handle_cache_;
206 std::unordered_map<AnfNode *, std::shared_ptr<std::vector<OutHandler>>> tuple_out_handle_cache_;
207 std::unordered_map<AnfNode *, std::shared_ptr<std::vector<AnfNodePtr>>> case_input_handle_cache_;
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/
Daicpu_kernel_build.cc40 using FNodeAttrHandle = std::function<void(const std::shared_ptr<AnfNode> &anf_node, mindspore::Nod…
42 bool SetIOIputSize(const std::shared_ptr<AnfNode> &anf_node, const size_t &input_num, in SetIOIputSize()
77 bool SetIOSize(const std::shared_ptr<AnfNode> &anf_node, const std::shared_ptr<AicpuOpKernelMod> &k… in SetIOSize()
144 void SetNodeAttr(const std::shared_ptr<AnfNode> &anf_node, mindspore::NodeDef *proto) { in SetNodeAttr()
179 void SetNodeInputs(const std::shared_ptr<AnfNode> &anf_node, mindspore::NodeDef *proto) { in SetNodeInputs()
219 void SetNodeOutputs(const std::shared_ptr<AnfNode> &anf_node, mindspore::NodeDef *proto) { in SetNodeOutputs()
249 void SetNodedefProto(const std::shared_ptr<AnfNode> &anf_node, mindspore::NodeDef *proto) { in SetNodedefProto()
266 bool CreateNodeDefBytes(const std::shared_ptr<AnfNode> &anf_node, in CreateNodeDefBytes()
295 const std::shared_ptr<AnfNode> &anf_node, size_t input_num) { in SetExtInfoInputShapeType()
335 const std::shared_ptr<AnfNode> &anf_node, size_t output_num) { in SetExtInfoOutputShapeType()
[all …]
Daicpu_kernel_build.h23 KernelModPtr AicpuOpBuild(const std::shared_ptr<AnfNode> &anf_node);
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/
Dprim_bprop_optimizer.h112 void ArgInfoRefresh(const std::shared_ptr<AnfNode> &param, ParamUsingInfo *arg_info) const;
114 void AnalysisNodeUsingInfo(const NodeUsersMap &node_users, const std::shared_ptr<AnfNode> &param,
119 void AalysisForTupleGetItem(const NodeUsersMap &node_users, const std::shared_ptr<AnfNode> &param,
Dprim_bprop_optimizer.cc76 const std::shared_ptr<AnfNode> &param, in AnalysisNodeUsingInfo()
107 const std::shared_ptr<AnfNode> &param, in AalysisForTupleGetItem()
143 void PrimBpropOptGraphLevel2Info::ArgInfoRefresh(const std::shared_ptr<AnfNode> &param, in ArgInfoRefresh()
/third_party/mindspore/mindspore/core/base/
Dbase.h130 class AnfNode;
131 using AnfNodePtr = std::shared_ptr<AnfNode>;
134 using AnfNodeWeakPtr = std::weak_ptr<AnfNode>;
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dpattern_engine.h92 class VarNode : public AnfNode {
94 VarNode(const VarPtr &value, const FuncGraphPtr &func_graph) : AnfNode(func_graph), var_(value) {} in VarNode()
96 MS_DECLARE_PARENT(VarNode, AnfNode);
/third_party/mindspore/tests/ut/cpp/utils/
Dbaseref_test.cc63 MS_LOG(INFO) << "anf typeid:" << dyn_cast<AnfNode>(anf).get(); in TEST_F()
67 ASSERT_EQ(isa<AnfNode>(d), true); in TEST_F()
/third_party/mindspore/mindspore/lite/tools/anf_exporter/
Danf_exporter.h58 int ConvertInputCNode(const std::shared_ptr<AnfNode> &input_anode, schema::CNodeT *output_cnode);
76 … bool keep_graph, bool copy_primitive, const std::shared_ptr<AnfNode> &partial_anode = nullptr);
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/
Dhost_kernel_build.h23 KernelModPtr HostOpBuild(const std::shared_ptr<AnfNode> &anf_node);
Dhost_kernel_build.cc26 KernelModPtr HostOpBuild(const std::shared_ptr<AnfNode> &anf_node) { in HostOpBuild()
/third_party/mindspore/tests/ut/cpp/stub/dynamic_shape/
Ddynamic_shape_stub.cc47 KernelModPtr HostOpBuild(const std::shared_ptr<AnfNode> &anf_node) { return nullptr; } in HostOpBuild()
/third_party/mindspore/mindspore/core/utils/
Dtrace_base.h38 std::string DumpSourceLines(AnfNode *node);
Dtrace_base.cc153 std::string DumpSourceLines(AnfNode *node) { in DumpSourceLines()
158 AnfNodePtr ptr = std::static_pointer_cast<AnfNode>(node->shared_from_this()); in DumpSourceLines()
Dinfo.h199 void set_node(const std::shared_ptr<AnfNode> &node) { node_ = AnfNodeWeakPtr(node); } in set_node()
200 std::shared_ptr<AnfNode> get_node() const { return node_.lock(); } in get_node()
/third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/
Dgather_actor.h113 std::vector<std::pair<size_t, AnfNode *>> device_tensor_store_keys_;

12