Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dsymbolic_shape_analysis.h22 std::shared_ptr<Graph> partial_eval_shape_graph, in ShapeComputeGraphMapping()
26 : partial_eval_shape_graph(std::move(partial_eval_shape_graph)), in ShapeComputeGraphMapping()
32 std::shared_ptr<Graph> partial_eval_shape_graph; member
H A Dsymbolic_shape_runtime_fusion.cpp35 shape_mapping.partial_eval_shape_graph->inputs()) { in InsertSymbolicShapesCompute()
56 *shape_mapping.partial_eval_shape_graph, in InsertSymbolicShapesCompute()
60 i < shape_mapping.partial_eval_shape_graph->outputs().size(); in InsertSymbolicShapesCompute()
62 Value* output = shape_mapping.partial_eval_shape_graph->outputs().at(i); in InsertSymbolicShapesCompute()
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_symbolic_shape_analysis.py531 g = shape_compute_graph.partial_eval_shape_graph()
553 g = shape_compute_graph.partial_eval_shape_graph()
566 g = shape_compute_graph.partial_eval_shape_graph()
677 g = shape_compute_graph.partial_eval_shape_graph()
/aosp_15_r20/external/pytorch/torch/csrc/jit/python/
H A Dpython_ir.cpp1146 return g.partial_eval_shape_graph; in initPythonIRBindings()