Searched defs:write_get_raw_stream (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ | ||
| H A D | cpp_wrapper_cuda.py | 189 def write_get_raw_stream(self, index, graph=None): member in CppWrapperCuda |
| H A D | wrapper.py | 690 def write_get_raw_stream(self, device_idx: int, graph=None) -> str: member in WrapperCodeGen |