Searched defs:node_json (Results 1 – 2 of 2) sorted by relevance
377 std::string AkgKernelJsonGenerator::GetTensorName(const nlohmann::json &node_json, const std::strin… in GetTensorName()440 void AkgKernelJsonGenerator::SaveNodeAddress(const AnfNodePtr &anf_node, nlohmann::json *node_json)… in SaveNodeAddress()458 …nelJsonGenerator::GenerateSingleKernelJson(const AnfNodePtr &anf_node, nlohmann::json *node_json) { in GenerateSingleKernelJson()497 bool AkgKernelJsonGenerator::GetIOSize(const nlohmann::json &node_json, std::vector<size_t> *input_… in GetIOSize()665 nlohmann::json node_json; in GenSingleJsons() local
734 …etInputTensorValue(const AnfNodePtr &anf_node, size_t input_idx, nlohmann::json *const node_json) { in GetInputTensorValue()