Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/passes/
Dsymbolic_shape_analysis.cpp291 for (size_t op_in_index = 0; in refineInputUnionTypes() local
292 op_in_index < shape_compute_graph_->inputs().size(); in refineInputUnionTypes()
293 op_in_index++) { in refineInputUnionTypes()
294 auto type = parent_graph_node->input(op_in_index)->type(); in refineInputUnionTypes()
296 .at(op_in_index) in refineInputUnionTypes()
303 .at(op_in_index) in refineInputUnionTypes()
307 .at(op_in_index) in refineInputUnionTypes()
310 shape_compute_graph_->inputs().at(op_in_index)->setType(type); in refineInputUnionTypes()
323 size_t op_in_index = 0; in substituteConstantInputs() local
324 while (op_in_index < shape_compute_graph_->inputs().size()) { in substituteConstantInputs()
[all …]