Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dsymbolic_shape_runtime_fusion.cpp24 Node* tensorexpr_graph) { in InsertSymbolicShapesCompute()
182 const std::shared_ptr<Graph>& tensorexpr_graph) { in TryGeneralizeInputDimensionsToSymbolicShapes()
216 const std::shared_ptr<Graph>& tensorexpr_graph) { in moveConstantTensorsOutOfSubgraph()
258 auto tensorexpr_graph = SubgraphUtils::getSubgraph(tensorexpr_graph_node); in GenerateGuard() local
/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_shape_analysis.cpp145 auto tensorexpr_graph = te_group->g(attr::Subgraph); in TEST() local