Home
last modified time | relevance | path

Searched refs:node_json (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/
Dakg_kernel_json_generator.cc377 std::string AkgKernelJsonGenerator::GetTensorName(const nlohmann::json &node_json, const std::strin… in GetTensorName() argument
379 if (node_json.count(tag) == 0) { in GetTensorName()
380 MS_LOG(ERROR) << "Node [" << node_json.dump() << "] has no key [" << tag << "]."; in GetTensorName()
384 auto const &tag_desc = node_json[tag]; in GetTensorName()
409 … const std::pair<size_t, size_t> &position, nlohmann::json *node_json) const { in SetTensorName()
410 MS_EXCEPTION_IF_NULL(node_json); in SetTensorName()
411 if (node_json->count(tag) == 0) { in SetTensorName()
412 MS_LOG(ERROR) << "Node [" << node_json->dump() << "] has no key [" << tag << "]."; in SetTensorName()
416 nlohmann::json *tag_desc = &((*node_json)[tag]); in SetTensorName()
440 void AkgKernelJsonGenerator::SaveNodeAddress(const AnfNodePtr &anf_node, nlohmann::json *node_json)… in SaveNodeAddress() argument
[all …]
Dakg_kernel_json_generator.h101 bool GenerateSingleKernelJson(const AnfNodePtr &anf_node, nlohmann::json *node_json);
123 bool GetIOSize(const nlohmann::json &node_json, std::vector<size_t> *input_size,
137 nlohmann::json *node_json) const;
138 std::string GetTensorName(const nlohmann::json &node_json, const std::string &tag,
146 void SaveNodeAddress(const AnfNodePtr &anf_node, nlohmann::json *node_json);
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dcommon_utils.cc734 …etInputTensorValue(const AnfNodePtr &anf_node, size_t input_idx, nlohmann::json *const node_json) { in GetInputTensorValue() argument
736 MS_EXCEPTION_IF_NULL(node_json); in GetInputTensorValue()
765 (*node_json)["value"] = static_cast<double *>(data)[0]; in GetInputTensorValue()
767 (*node_json)["value"] = static_cast<float *>(data)[0]; in GetInputTensorValue()
770 (*node_json)["value"] = static_cast<float>(val[0]); in GetInputTensorValue()
772 (*node_json)["value"] = static_cast<uint64_t *>(data)[0]; in GetInputTensorValue()
774 (*node_json)["value"] = static_cast<uint32_t *>(data)[0]; in GetInputTensorValue()
776 (*node_json)["value"] = static_cast<uint16_t *>(data)[0]; in GetInputTensorValue()
778 (*node_json)["value"] = static_cast<uint8_t *>(data)[0]; in GetInputTensorValue()
780 (*node_json)["value"] = static_cast<int64_t *>(data)[0]; in GetInputTensorValue()
[all …]
Dcommon_utils.h100 …GetInputTensorValue(const AnfNodePtr &anf_node, size_t input_idx, nlohmann::json *const node_json);
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/
Dtbe_json_creator.cc249 for (auto node_json : (*kernel_json)[kJOpList]) { in GenFusionOpName()
250 if (GetJsonValue<std::string>(node_json, kJType) != kJData) { in GenFusionOpName()
251 json_name_.append(node_json[kJFuncName]); in GenFusionOpName()
288 for (auto &node_json : (*kernel_json).at(kJOpList)) { in AddOpNameForComputeNode()
290 if (GetJsonValue<std::string>(node_json, kJType) != kJData) { in AddOpNameForComputeNode()
291 node_json[kJOpName] = op_name; in AddOpNameForComputeNode()