Searched refs:kJsonKeyOutputDesc (Results 1 – 4 of 4) sorted by relevance
386 if (tag == kJsonKeyOutputDesc) { in GetTensorName()418 if (tag == kJsonKeyOutputDesc) { in SetTensorName()484 (*node_json)[kJsonKeyOutputDesc] = outputs_json; in GenerateSingleKernelJson()517 for (size_t i = 0; i < node_json[kJsonKeyOutputDesc].size(); i++) { in GetIOSize()518 std::string dtype = node_json[kJsonKeyOutputDesc][i][kJsonKeyDataType]; in GetIOSize()521 std::accumulate(node_json[kJsonKeyOutputDesc][i][kJsonKeyShape].begin(), in GetIOSize()522 … node_json[kJsonKeyOutputDesc][i][kJsonKeyShape].end(), nbyte, std::multiplies<size_t>()); in GetIOSize()570 auto name = GetTensorName((*node_json_map)[anf_node], kJsonKeyOutputDesc, {0, 0}); in GenStitchJson()612 (*kernel_json)[kJsonKeyOutputDesc] = in CollectFusedJson()693 …GetTensorName((*node_json_map)[tmp_input.first], kJsonKeyOutputDesc, std::make_pair(0, tmp_input.s… in UpdateTensorName()[all …]
200 std::vector<nlohmann::json> output_descs = cnode_json[kJsonKeyOutputDesc]; in DecodeOutputDesc()419 std::vector<nlohmann::json> output_descs = kernel_json[kJsonKeyOutputDesc]; in DecodeFusedNodes()
39 constexpr auto kJsonKeyOutputDesc = "output_desc"; variable
61 std::vector<nlohmann::json> output_descs = op_desc[kJsonKeyOutputDesc]; in IsRecomputeOp()102 std::vector<nlohmann::json> output_descs = op_desc[kJsonKeyOutputDesc]; in SetStitchAttr()