Searched defs:BoxedBool (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/ |
| H A D | autograd_cache.py | 168 fx_config: Dict[str, BoxedBool], argument
232 fx_config: Dict[str, BoxedBool], argument
255 def load(self, example_inputs, fx_config: Dict[str, BoxedBool]) -> CompiledFxGraph: argument
321 fx_config: Dict[str, BoxedBool], argument
455 cudagraphs: BoxedBool,
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | compile_fx.py | 440 cudagraphs: Optional[BoxedBool] = None,
690 cudagraphs: Optional[BoxedBool] = None,
1184 cudagraphs: BoxedBool,
|
| H A D | utils.py | 1766 class BoxedBool: class
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | aot_autograd.py | 945 cudagraphs: Optional[BoxedBool] = None,
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_subclasses.py | 1779 cudagraphs: Optional[BoxedBool] = None,
|