Home
last modified time | relevance | path

Searched refs:enclosing_graph_input (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dsymbolic_shape_runtime_fusion.cpp36 auto enclosing_graph_input = in InsertSymbolicShapesCompute() local
39 enclosing_graph_input != in InsertSymbolicShapesCompute()
41 if (*enclosing_graph_input->second->type() == *shape_graph_input->type()) { in InsertSymbolicShapesCompute()
43 enclosing_graph_input->second->offset())); in InsertSymbolicShapesCompute()
46 enclosing_graph_input->second->type()->cast<TensorType>() && in InsertSymbolicShapesCompute()
51 enclosing_graph_input->second->offset())})); in InsertSymbolicShapesCompute()