Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dsymbolic_shape_analysis.cpp282 std::shared_ptr<Graph> shape_compute_graph_; member
292 op_in_index < shape_compute_graph_->inputs().size(); in refineInputUnionTypes()
295 if (auto opt_type = shape_compute_graph_->inputs() in refineInputUnionTypes()
302 shape_compute_graph_->inputs() in refineInputUnionTypes()
306 } else if (shape_compute_graph_->inputs() in refineInputUnionTypes()
310 shape_compute_graph_->inputs().at(op_in_index)->setType(type); in refineInputUnionTypes()
317 if (shape_compute_graph_->inputs().empty()) { in substituteConstantInputs()
324 while (op_in_index < shape_compute_graph_->inputs().size()) { in substituteConstantInputs()
325 Value* graph_in_var = shape_compute_graph_->inputs().at(op_in_index); in substituteConstantInputs()
357 auto new_inp = shape_compute_graph_->insertInput(op_in_index + j); in substituteConstantInputs()
[all …]