Searched refs:capture_end (Results 1 – 20 of 20) sorted by relevance
75 def capture_end(self): member in CUDAGraph84 super().capture_end()186 self.cuda_graph.capture_end()
34 void capture_end();
153 void CUDAGraph::capture_end() { in capture_end() function in at::cuda::CUDAGraph
58 torch::wrap_pybind_function_no_gil(&at::cuda::CUDAGraph::capture_end)) in THCPGraph_init()
1578 g.capture_end()1594 g.capture_end()1665 g.capture_end()1695 graph.capture_end()1771 g.capture_end()1785 g.capture_end()1813 g.capture_end()1868 g.capture_end()1944 g.capture_end()2065 g.capture_end()[all …]
948 LambdaExpr::capture_iterator LambdaExpr::capture_end() const { in capture_end() function in LambdaExpr953 return capture_range(capture_begin(), capture_end()); in captures()975 return capture_end(); in implicit_capture_end()
3021 E = LE->capture_end(); I != E; ++I) in HasSideEffects()
2141 return capture_range(capture_begin(), capture_end()); in captures()2144 return capture_const_range(capture_begin(), capture_end()); in captures()2153 capture_iterator capture_end() const { in capture_end() function
3584 capture_const_iterator capture_end() const { return captures().end(); } in capture_end() function
1629 capture_iterator capture_end() const;
183 E = Lambda->capture_end(); in TraverseLambdaExpr()
10154 CEnd = E->capture_end(); in TransformLambdaExpr()10251 CEnd = E->capture_end(); in TransformLambdaExpr()
142 if (Lambda->capture_begin() == Lambda->capture_end()) in VisitLambdaExpr()
209 auto CE = BD->capture_end(); in VisitBlockExpr()
975 :meth:`CUDAGraph.capture_end<torch.cuda.CUDAGraph.capture_end>` calls.
342 (block->capture_end() - block->capture_begin())); in computeBlockInfo()
188 E = S.capture_end(); in Kind()
5727 CE = CS.capture_end(); in emitTargetCall()
326 Record.push_back(std::distance(S->capture_begin(), S->capture_end())); in VisitCapturedStmt()
2014 def capture_end(self) -> None: ...