Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/passes/
Dsymbolic_shape_analysis.cpp819 auto stitched_shape_compute_graph = std::make_shared<Graph>(); in run() local
864 curr, partial_eval_graph, stitched_shape_compute_graph); in run()
872 made_change = shapeGraphCleanupPasses(stitched_shape_compute_graph); in run()
881 for (size_t i = 0; i < stitched_shape_compute_graph->outputs().size(); in run()
883 Value* output = stitched_shape_compute_graph->outputs().at(i); in run()
897 stitched_shape_compute_graph->eraseOutput(erase_indices[i]); in run()
899 for (size_t i = 0; i < stitched_shape_compute_graph->inputs().size();) { in run()
900 if (!stitched_shape_compute_graph->inputs().at(i)->hasUses()) { in run()
902 stitched_shape_compute_graph->inputs().at(i)); in run()
903 stitched_shape_compute_graph->eraseInput(i); in run()
[all …]