Home
last modified time | relevance | path

Searched defs:FunctionalGraphSlicer (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dcreate_functional_graphs.cpp15 struct FunctionalGraphSlicer { struct
16 FunctionalGraphSlicer(std::shared_ptr<Graph> graph) in FunctionalGraphSlicer() argument
19 void run() { in run()
36 bool isEmptyFunctionalGraph(Node* n) { in isEmptyFunctionalGraph()
41 void nonConstNodes(Block* block, size_t* num) { in nonConstNodes()
56 bool inlineIfTooSmall(Node* n) { in inlineIfTooSmall()
68 bool CreateFunctionalGraphsImpl(Block* block) { in CreateFunctionalGraphsImpl()
130 bool AnalyzeFunctionalSubset(Node* n) { in AnalyzeFunctionalSubset()
164 void AnalyzeFunctionalSubset(at::ArrayRef<Block*> blocks) { in AnalyzeFunctionalSubset()
170 bool AnalyzeFunctionalSubset(Block* block) { in AnalyzeFunctionalSubset()
[all …]