Searched refs:run_with_cuda_stream (Results 1 – 2 of 2) sorted by relevance
24 std::vector<at::Tensor> run_with_cuda_stream(
26 std::vector<at::Tensor> AOTIModelContainerRunnerCuda::run_with_cuda_stream( in run_with_cuda_stream() function in torch::inductor::AOTIModelContainerRunnerCuda