Searched defs:CppCSEVariable (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | cpp.py | 2262 def _get_mask_cast(self, mask: CppCSEVariable, dtype: torch.dtype) -> str:
2275 load_mask: Optional[CppCSEVariable] = None,
2313 store_value: Optional[Union[str, CppCSEVariable]] = None, argument
2350 def vec_to_array(vec_var: CppCSEVariable) -> CppCSEVariable:
2471 value: Union[str, CppCSEVariable], argument
2742 def broadcast(self, scalar_var: CppCSEVariable) -> CppCSEVariable:
2760 def arange(self, index: CppCSEVariable, stride: sympy.Symbol) -> CppCSEVariable:
|
| H A D | cpp_utils.py | 173 class CppCSEVariable(CSEVariable): class
|