Searched refs:input_src_node (Results 1 – 1 of 1) sorted by relevance
686 auto input_src_node = n->input(node_index)->node(); in getNodeInputShapes() local687 if (input_src_node->kind() == prim::ListConstruct && in getNodeInputShapes()726 if (input_src_node->kind() == aten::size && in getNodeInputShapes()728 auto ten_inp = input_src_node->input(); in getNodeInputShapes()