Home
last modified time | relevance | path

Searched full:is_root_operator (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/pytorch/test/edge/
H A Dselected_operators.yaml18 is_root_operator: true
24 is_root_operator: true
30 is_root_operator: true
36 is_root_operator: true
42 is_root_operator: true
48 is_root_operator: true
54 is_root_operator: true
60 is_root_operator: true
66 is_root_operator: true
72 is_root_operator: true
[all …]
/aosp_15_r20/external/pytorch/tools/test/
H A Dtest_selective_build.py15 is_root_operator=True,
20 self.assertTrue(op.is_root_operator)
32 is_root_operator: Yes
36 is_root_operator: No
40 is_root_operator: No
51 is_root_operator: Yes
57 is_root_operator: No
100 self.assertFalse(selector_legacy_v1.is_root_operator("aten::add"))
111 self.assertTrue(selector_legacy_v1.is_root_operator("aten::add"))
115 self.assertTrue(selector_legacy_v1.is_root_operator("aten::add.float"))
[all …]
/aosp_15_r20/external/pytorch/torchgen/selective_build/
H A Doperator.py28 is_root_operator: bool
58 "is_root_operator",
74 is_root_operator = op_info.get("is_root_operator", True)
75 assert isinstance(is_root_operator, bool)
91 is_root_operator=is_root_operator,
103 is_root_operator=True,
111 "is_root_operator": self.is_root_operator,
145 is_root_operator=lhs.is_root_operator or rhs.is_root_operator,
H A Dselector.py155 allow_list: set[str], is_root_operator: bool, is_used_for_training: bool
161 "is_root_operator": is_root_operator,
193 is_root_operator=False,
215 def is_root_operator(self, name: str) -> bool: member in SelectiveBuilder
223 return op.is_root_operator
228 return base_op.include_all_overloads and base_op.is_root_operator
/aosp_15_r20/external/pytorch/tools/code_analyzer/
H A Dgen_operators_yaml.py41 # is_root_operator: Yes
48 # is_root_operator: No
354 "is_root_operator": True,
368 "is_root_operator": False,
382 "is_root_operator": True,
396 "is_root_operator": False,
412 "is_root_operator": True,
426 "is_root_operator": False,
440 "is_root_operator": True,
454 "is_root_operator": False,
/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/model_tracer/
H A Dtracer.cpp61 bool is_root_operator, in printOpYAML() argument
68 << "is_root_operator: " << (is_root_operator ? "true" : "false") << '\n'; in printOpYAML()
78 bool is_root_operator, in printOpsYAML() argument
81 printOpYAML(out, 2, it, false, is_root_operator, false); in printOpsYAML()
181 true /* is_root_operator */, in main()
187 false /* is_root_operator */, in main()
/aosp_15_r20/external/executorch/codegen/tools/
H A Dgen_oplist.py32 # is_root_operator: Yes
39 # is_root_operator: No
168 "is_root_operator": True,
/aosp_15_r20/external/executorch/codegen/tools/test/
H A Dtest_gen_selected_op_variants.py32 is_root_operator: Yes
36 is_root_operator: No
/aosp_15_r20/external/pytorch/tools/setup_helpers/
H A Dgenerate_code.py94 is_root_operator = True
101 is_root_operator,
/aosp_15_r20/external/pytorch/tools/jit/
H A Dgen_unboxing.py36 if not self.selector.is_root_operator(f"aten::{f.func.name}"):
100 if not self.selector.is_root_operator(f"aten::{f.func.name}"):
/aosp_15_r20/external/pytorch/torchgen/
H A Dgen_executorch.py132 if not self.selector.is_root_operator(f"{f.namespace}::{f.func.name}"):
194 if not self.selector.is_root_operator(op_name):
/aosp_15_r20/external/pytorch/tools/lite_interpreter/
H A Dgen_selected_mobile_ops_header.py55 if op.is_root_operator: