Home
last modified time | relevance | path

Searched refs:compute_json (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/
Dtbe_json_creator.cc221 bool TbeJsonCreator::GenComputeJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json) { in GenComputeJson() argument
223 MS_EXCEPTION_IF_NULL(compute_json); in GenComputeJson()
226 if (!GenInputsJson(anf_node, compute_json)) { in GenComputeJson()
230 if (!GenOutputsJson(anf_node, compute_json)) { in GenComputeJson()
234 GenOutputDataDescJson(anf_node, compute_json); in GenComputeJson()
235 GenAttrsDescJson(anf_node, compute_json); in GenComputeJson()
236 GenComputeCommonJson(anf_node, compute_json); in GenComputeJson()
237 GenOtherJson(anf_node, compute_json); in GenComputeJson()
343 void TbeJsonCreator::GenAttrsDescJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json) { in GenAttrsDescJson() argument
345 MS_EXCEPTION_IF_NULL(compute_json); in GenAttrsDescJson()
[all …]
Dtbe_json_creator.h75 bool GenComputeJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json);
76 …virtual bool GenInputsJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json) { return fals… in GenInputsJson() argument
77 …virtual bool GenOutputsJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json) { return fal… in GenOutputsJson() argument
78 void GenOutputDataDescJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json);
79 void GenComputeCommonJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json);
80 virtual void GenOtherJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json) {} in GenOtherJson() argument
81 void GenAttrsDescJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json);
Dsingle_tbe_json_creator.cc58 nlohmann::json compute_json; in GenOpListJson() local
59 if (!GenComputeJson(anf_node, &compute_json)) { in GenOpListJson()
63 GenDataJson(anf_node, compute_json, op_list_json); in GenOpListJson()
64 (*op_list_json).push_back(compute_json); in GenOpListJson()
69 …d SingleTbeJsonCreator::GenDataJson(const AnfNodePtr &anf_node, const nlohmann::json &compute_json, in GenDataJson() argument
77 auto inputs_json = GetJsonValue<std::vector<nlohmann::json>>(compute_json, kJInputDesc); in GenDataJson()
99 bool SingleTbeJsonCreator::GenInputsJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json) { in GenInputsJson() argument
101 MS_EXCEPTION_IF_NULL(compute_json); in GenInputsJson()
110 (*compute_json)[kJInputDesc] = inputs_json; in GenInputsJson()
142 (*compute_json)[kJInputDesc] = inputs_json; in GenInputsJson()
[all …]
Dfusion_tbe_json_creator.cc73 nlohmann::json compute_json; in GenOpListJson() local
74 if (!GenComputeJson(compute_node, &compute_json)) { in GenOpListJson()
79 compute_json[kJOriName] = {fusion_scope_info.full_name}; in GenOpListJson()
80 compute_list.push_back(compute_json); in GenOpListJson()
124 const std::vector<nlohmann::json> &compute_json, in GenDataJson() argument
133 auto inputs_desc = GetJsonValue<std::vector<nlohmann::json>>(compute_json.at(i), kJInputDesc); in GenDataJson()
170 …usionBuildTbeJsonCreator::GenInputsJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json) { in GenInputsJson() argument
172 MS_EXCEPTION_IF_NULL(compute_json); in GenInputsJson()
217 (*compute_json)[kJInputDesc] = input_desc_list; in GenInputsJson()
242 …sionBuildTbeJsonCreator::GenOutputsJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json) { in GenOutputsJson() argument
[all …]
Dfusion_tbe_json_creator.h32 bool GenInputsJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json) override;
33 bool GenOutputsJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json) override;
37 …Json(const std::vector<AnfNodePtr> &compute_nodes, const std::vector<nlohmann::json> &compute_json,
41 void GenOtherJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json) override;
Dsingle_tbe_json_creator.h31 void GenDataJson(const AnfNodePtr &anf_node, const nlohmann::json &compute_json,
33 bool GenInputsJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json) override;
39 bool GenOutputsJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json) override;
43 void GenOtherJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json) override;