Searched refs:SelectiveBuildOperator (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torchgen/selective_build/ |
| H A D | operator.py | 13 class SelectiveBuildOperator: class
55 ) -> SelectiveBuildOperator:
89 return SelectiveBuildOperator(
100 ) -> SelectiveBuildOperator:
101 return SelectiveBuildOperator(
133 lhs: SelectiveBuildOperator, rhs: SelectiveBuildOperator argument
134 ) -> SelectiveBuildOperator:
140 return SelectiveBuildOperator(
156 lhs: dict[str, SelectiveBuildOperator], argument
157 rhs: dict[str, SelectiveBuildOperator],
[all …]
|
| H A D | selector.py | 13 SelectiveBuildOperator,
39 operators: dict[str, SelectiveBuildOperator]
108 operators[k] = SelectiveBuildOperator.from_yaml_dict(k, v)
191 not_training_op = SelectiveBuildOperator(
222 op: SelectiveBuildOperator = self.operators[name]
227 base_op: SelectiveBuildOperator = self.operators[name]
|
| /aosp_15_r20/external/pytorch/tools/code_analyzer/ |
| H A D | gen_operators_yaml.py | 19 SelectiveBuildOperator,
351 op = SelectiveBuildOperator.from_yaml_dict(
365 op = SelectiveBuildOperator.from_yaml_dict(
379 op = SelectiveBuildOperator.from_yaml_dict(
393 op = SelectiveBuildOperator.from_yaml_dict(
409 op = SelectiveBuildOperator.from_yaml_dict(
423 op = SelectiveBuildOperator.from_yaml_dict(
437 op = SelectiveBuildOperator.from_yaml_dict(
451 op = SelectiveBuildOperator.from_yaml_dict(
465 operators: dict[str, SelectiveBuildOperator] = {}
|
| /aosp_15_r20/external/pytorch/tools/test/ |
| H A D | test_selective_build.py | 13 op = SelectiveBuildOperator(
136 op1 = SelectiveBuildOperator(
143 op2 = SelectiveBuildOperator(
150 op3 = SelectiveBuildOperator(
157 op4 = SelectiveBuildOperator(
|
| /aosp_15_r20/external/executorch/codegen/tools/ |
| H A D | gen_oplist.py | 18 from torchgen.selective_build.operator import SelectiveBuildOperator
165 op = SelectiveBuildOperator.from_yaml_dict(
|