Searched defs:stream_class (Results 1 – 2 of 2) sorted by relevance
152 auto stream_class = m.class_<torch::jit::CUDAStream>("Stream").def( in TORCH_LIBRARY() local
220 jclass stream_class; member228 jclass stream_class; member