Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dsymbolic_shape_runtime_fusion.cpp33 std::vector<Value*> shape_compute_graph_inputs; in InsertSymbolicShapesCompute() local
42 shape_compute_graph_inputs.push_back(tensorexpr_graph->inputs().at( in InsertSymbolicShapesCompute()
48 shape_compute_graph_inputs.push_back(enclosing_graph->insert( in InsertSymbolicShapesCompute()
57 shape_compute_graph_inputs); in InsertSymbolicShapesCompute()