Home
last modified time | relevance | path

Searched defs:attrs_json (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_adapter.cc195 nlohmann::json *attrs_json) { in RunAttrPass()
208 nlohmann::json *attrs_json) { in MaxiOrMinimumGradAttrJsonPass()
234 nlohmann::json *attrs_json) { in CastAttrJsonPass()
435 nlohmann::json *attrs_json) { in CastAttrJsonPrePass()
461 void TbeAdapter::CastAttrJsonPost(const AnfNodePtr &anf_node, nlohmann::json *attrs_json) { in CastAttrJsonPost()
475 void TbeAdapter::LayerNormAttrJsonPost(const AnfNodePtr &anf_node, nlohmann::json *attrs_json) { in LayerNormAttrJsonPost()
Dtbe_kernel_build.cc219 nlohmann::json attrs_json; in GenTbeSingleKernelJson() local
545 … const std::shared_ptr<OpInfo> &op_info, nlohmann::json *attrs_json) { in GenTbeAttrJson()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/
Dtbe_json_creator.cc296 …r::GenAttrsJson(const AnfNodePtr &anf_node, const OpInfoPtr &op_info, nlohmann::json *attrs_json) { in GenAttrsJson()
350 nlohmann::json attrs_json; in GenAttrsDescJson() local
532 nlohmann::json *attrs_json) { in AttrsJsonPreProcessing()
557 nlohmann::json *attrs_json) { in AttrsJsonPostProcessing()
Dfusion_tbe_json_creator.cc304 nlohmann::json *attrs_json) { in AttrsJsonPostProcessing()
Dsingle_tbe_json_creator.cc371 nlohmann::json *attrs_json) { in AttrsJsonPostProcessing()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/
Dakg_kernel_json_generator.cc299 nlohmann::json *attrs_json) { in CreateAttrDescJson()
488 nlohmann::json attrs_json; in GenerateSingleKernelJson() local
Dakg_kernel_json_decoder.cc148 bool DecodeAttrs(const nlohmann::json &attrs_json) { in DecodeAttrs()