Searched defs:is_inference (Results 1 – 12 of 12) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | graph.py | 308 is_inference=False, argument
431 def decide_layout_opt(gm, *, is_inference) -> bool: argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/templates/ |
| H A D | Functions.h | 123 inline bool is_inference(const Tensor& tensor) { in is_inference() function
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/backends/ |
| H A D | cudagraphs.py | 129 def forward_cudagraphs(aot_model, aot_inputs, is_inference=False): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TypeProperties.cpp | 39 bool is_inference(const Tensor& self) { in is_inference() function
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | cpp_wrapper_cuda.py | 203 def generate(self, is_inference): argument
|
| H A D | wrapper.py | 855 def generate(self, is_inference): argument
859 def _generate(self, is_inference): argument
|
| H A D | cpp_wrapper_cpu.py | 903 def generate(self, is_inference): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | TensorBase.h | 541 bool is_inference() const { in is_inference() function
|
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | meta_utils.py | 464 is_inference: bool = False variable in MetaTensorDesc
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_torchinductor_dynamic_shapes.py | 894 def generate(self, is_inference, *args, **kwargs): argument
|
| H A D | test_cudagraph_trees.py | 185 self, *args, is_inference=True, is_backward=False, **kwargs argument
|
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | TensorImpl.h | 1233 bool is_inference() { in is_inference() function
|