Home
last modified time | relevance | path

Searched refs:TraceableTritonKernelWrapper (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_library/
H A Dtriton.py225 from torch._higher_order_ops.triton_kernel_wrap import TraceableTritonKernelWrapper
233 return TraceableTritonKernelWrapper(triton_kernel, None, None)
/aosp_15_r20/external/pytorch/torch/_higher_order_ops/
H A Dtriton_kernel_wrap.py1016 class TraceableTritonKernelWrapper: class