Home
last modified time | relevance | path

Searched defs:output_used_nums (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/
Dfusion_tbe_json_creator.cc250 … auto output_used_nums = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(anf_node, kAttrOutputUsedNum); in GenOutputsJson() local
292 …size_t> FusionBuildTbeJsonCreator::GetDescOutputIndex(const std::vector<int64_t> &output_used_nums in GetDescOutputIndex()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_build.cc1426 std::vector<size_t> TbeKernelBuild::GetDescOutputIndex(const std::vector<int64_t> &output_used_nums in GetDescOutputIndex()
1445 auto output_used_nums = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(cnode, kAttrOutputUsedNum); in GenFusionComputeOutputJson() local