Home
last modified time | relevance | path

Searched full:is_used_for_training (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/test/edge/
H A Dselected_operators.yaml19 is_used_for_training: true
25 is_used_for_training: true
31 is_used_for_training: true
37 is_used_for_training: true
43 is_used_for_training: true
49 is_used_for_training: true
55 is_used_for_training: true
61 is_used_for_training: true
67 is_used_for_training: true
73 is_used_for_training: true
[all …]
/aosp_15_r20/external/pytorch/tools/test/
H A Dtest_selective_build.py16 is_used_for_training=False,
21 self.assertFalse(op.is_used_for_training)
31 is_used_for_training: No
35 is_used_for_training: Yes
39 is_used_for_training: Yes
50 is_used_for_training: No
56 is_used_for_training: Yes
139 is_used_for_training=False,
146 is_used_for_training=False,
153 is_used_for_training=False,
[all …]
/aosp_15_r20/external/pytorch/torchgen/selective_build/
H A Doperator.py35 is_used_for_training: bool
59 "is_used_for_training",
77 is_used_for_training = op_info.get("is_used_for_training", True)
78 assert isinstance(is_used_for_training, bool)
92 is_used_for_training=is_used_for_training,
104 is_used_for_training=True,
112 "is_used_for_training": self.is_used_for_training,
149 is_used_for_training=lhs.is_used_for_training or rhs.is_used_for_training,
H A Dselector.py155 allow_list: set[str], is_root_operator: bool, is_used_for_training: bool
162 "is_used_for_training": is_used_for_training,
194 is_used_for_training=False,
207 return op.is_used_for_training or (
208 base_op.include_all_overloads and base_op.is_used_for_training
/aosp_15_r20/external/pytorch/tools/code_analyzer/
H A Dgen_operators_yaml.py42 # is_used_for_training: Yes
49 # is_used_for_training: No
355 "is_used_for_training": False,
369 "is_used_for_training": False,
383 "is_used_for_training": True,
397 "is_used_for_training": True,
413 "is_used_for_training": False,
427 "is_used_for_training": False,
441 "is_used_for_training": True,
455 "is_used_for_training": True,
H A Dgen_oplist.py30 if op.is_used_for_training:
/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/model_tracer/
H A Dtracer.cpp60 bool is_used_for_training, in printOpYAML() argument
65 << "is_used_for_training: " << (is_used_for_training ? "true" : "false") in printOpYAML()
77 bool is_used_for_training, in printOpsYAML() argument
180 false /* is_used_for_training */, in main()
186 false /* is_used_for_training */, in main()
/aosp_15_r20/external/executorch/codegen/tools/
H A Dgen_oplist.py33 # is_used_for_training: Yes
40 # is_used_for_training: No
169 "is_used_for_training": True,
/aosp_15_r20/external/executorch/codegen/tools/test/
H A Dtest_gen_selected_op_variants.py33 is_used_for_training: Yes
37 is_used_for_training: No
/aosp_15_r20/external/pytorch/tools/setup_helpers/
H A Dgenerate_code.py95 is_used_for_training = True
102 is_used_for_training,