Home
last modified time | relevance | path

Searched full:root_operators (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dupdate_production_ops.py11 root_operators = {} variable
18 for op in info["root_operators"]:
20 root_operators[op] = 1 + (root_operators[op] if op in root_operators else 0)
36 root_operators = { variable
37 x: root_operators[x] for x in root_operators if x.split("::")[0] in namespaces
45 yaml.safe_dump({"root_operators": root_operators}, f)
H A Dgen_test_model.py115 production_ops = set(production_ops_dict["root_operators"].keys())
122 total_occurances = sum(production_ops_dict["root_operators"].values())
124 op: production_ops_dict["root_operators"][op] for op in covered_ops
127 op: production_ops_dict["root_operators"][op] for op in uncovered_ops
H A Dmodel_ops.yaml1 root_operators:
/aosp_15_r20/external/pytorch/tools/test/
H A Dgen_operators_yaml_test.py63 "root_operators": [],
73 "root_operators": [],
82 "root_operators": [],
92 "root_operators": [],
116 "root_operators": [],
126 "root_operators": [],
153 "root_operators": [],
163 "root_operators": [],
/aosp_15_r20/external/pytorch/tools/code_analyzer/
H A Dgen_operators_yaml.py281 static_root_ops = static_root_ops | set(model_info["root_operators"])
292 model_info["root_operators"]
299 model_info["root_operators"]