Searched refs:FusionBuildTbeJsonCreator (Results 1 – 4 of 4) sorted by relevance
33 bool FusionBuildTbeJsonCreator::GenJson(const FusionScopeInfo &fusion_scope_info, nlohmann::json *f… in GenJson()57 bool FusionBuildTbeJsonCreator::GenOpListJson(const FusionScopeInfo &fusion_scope_info, in GenOpListJson()94 bool FusionBuildTbeJsonCreator::CheckInput(const FusionScopeInfo &fusion_scope_info) { in CheckInput()123 void FusionBuildTbeJsonCreator::GenDataJson(const std::vector<AnfNodePtr> &compute_nodes, in GenDataJson()155 AnfNodePtr FusionBuildTbeJsonCreator::GetInputCNode(const AnfNodePtr &node, const nlohmann::json &i… in GetInputCNode()170 bool FusionBuildTbeJsonCreator::GenInputsJson(const AnfNodePtr &anf_node, nlohmann::json *compute_j… in GenInputsJson()221 bool FusionBuildTbeJsonCreator::CheckDynamicInput(const CNodePtr &cnode) { in CheckDynamicInput()242 bool FusionBuildTbeJsonCreator::GenOutputsJson(const AnfNodePtr &anf_node, nlohmann::json *compute_… in GenOutputsJson()281 void FusionBuildTbeJsonCreator::GenReusedOutputDesc(const AnfNodePtr &anf_node, size_t index, size_… in GenReusedOutputDesc()292 std::vector<size_t> FusionBuildTbeJsonCreator::GetDescOutputIndex(const std::vector<int64_t> &outpu… in GetDescOutputIndex()[all …]
24 class FusionBuildTbeJsonCreator : public TbeJsonCreator {26 FusionBuildTbeJsonCreator() : optional_index_(0) {} in FusionBuildTbeJsonCreator() function27 ~FusionBuildTbeJsonCreator() override = default;
307 auto tbe_json_creator = std::make_shared<FusionBuildTbeJsonCreator>(); in TEST_F()366 auto tbe_json_creator = std::make_shared<FusionBuildTbeJsonCreator>(); in TEST_F()
662 auto json_creator = std::make_shared<FusionBuildTbeJsonCreator>(); in AscendFusionOpCompile()