Searched refs:create_wait (Results 1 – 2 of 2) sorted by relevance
6440 ir._WaitKernel.create_wait(_c10d_functional.wait_tensor.default, inp)
8366 def create_wait(cls, kernel, inp: TensorBox) -> None: member in _WaitKernel