Searched refs:get_node_attr_from_input_ai (Results 1 – 5 of 5) sorted by relevance
224 static std::vector<int> get_node_attr_from_input_ai(const onnx::TensorProto& tp) in get_node_attr_from_input_ai() function385 shape = get_node_attr_from_input_ai(weights[node->input(1)]); in fuse_weight_reshape()505 shape = get_node_attr_from_input_ai(weights[node->input(1)]); in fuse_shufflechannel()559 shape3 = get_node_attr_from_input_ai(weights[node3->input(1)]); in fuse_shufflechannel()1258 shape = get_node_attr_from_input_ai(weights[node->input(1)]); in fuse_groupnorm()1761 shape = get_node_attr_from_input_ai(weights[node->input(1)]); in fuse_pixelshuffle()1882 shape = get_node_attr_from_input_ai(weights[node->input(1)]); in fuse_reorg()2072 shape = get_node_attr_from_input_ai(weights[node3->input(1)]); in fuse_lstm_gru_rnn()5174 pads = get_node_attr_from_input_ai(weights[node.input(1)]); in main()5327 shape = get_node_attr_from_input_ai(weights[node.input(1)]); in main()[all …]
224 static std::vector<int> get_node_attr_from_input_ai(const onnx::TensorProto& tp) in get_node_attr_from_input_ai() function385 shape = get_node_attr_from_input_ai(weights[node->input(1)]); in fuse_weight_reshape()505 shape = get_node_attr_from_input_ai(weights[node->input(1)]); in fuse_shufflechannel()559 shape3 = get_node_attr_from_input_ai(weights[node3->input(1)]); in fuse_shufflechannel()1258 shape = get_node_attr_from_input_ai(weights[node->input(1)]); in fuse_groupnorm()1761 shape = get_node_attr_from_input_ai(weights[node->input(1)]); in fuse_pixelshuffle()1882 shape = get_node_attr_from_input_ai(weights[node->input(1)]); in fuse_reorg()2072 shape = get_node_attr_from_input_ai(weights[node3->input(1)]); in fuse_lstm_gru_rnn()5168 pads = get_node_attr_from_input_ai(weights[node.input(1)]); in main()5321 shape = get_node_attr_from_input_ai(weights[node.input(1)]); in main()[all …]
224 static std::vector<int> get_node_attr_from_input_ai(const onnx::TensorProto& tp) in get_node_attr_from_input_ai() function385 shape = get_node_attr_from_input_ai(weights[node->input(1)]); in fuse_weight_reshape()505 shape = get_node_attr_from_input_ai(weights[node->input(1)]); in fuse_shufflechannel()559 shape3 = get_node_attr_from_input_ai(weights[node3->input(1)]); in fuse_shufflechannel()1258 shape = get_node_attr_from_input_ai(weights[node->input(1)]); in fuse_groupnorm()1761 shape = get_node_attr_from_input_ai(weights[node->input(1)]); in fuse_pixelshuffle()1882 shape = get_node_attr_from_input_ai(weights[node->input(1)]); in fuse_reorg()2079 shape = get_node_attr_from_input_ai(weights[node3->input(1)]); in fuse_lstm_gru_rnn()5249 pads = get_node_attr_from_input_ai(weights[node.input(1)]); in main()5402 shape = get_node_attr_from_input_ai(weights[node.input(1)]); in main()[all …]
224 static std::vector<int> get_node_attr_from_input_ai(const onnx::TensorProto& tp) in get_node_attr_from_input_ai() function385 shape = get_node_attr_from_input_ai(weights[node->input(1)]); in fuse_weight_reshape()505 shape = get_node_attr_from_input_ai(weights[node->input(1)]); in fuse_shufflechannel()559 shape3 = get_node_attr_from_input_ai(weights[node3->input(1)]); in fuse_shufflechannel()1254 shape = get_node_attr_from_input_ai(weights[node->input(1)]); in fuse_groupnorm()1568 shape = get_node_attr_from_input_ai(weights[node->input(1)]); in fuse_pixelshuffle()1689 shape = get_node_attr_from_input_ai(weights[node->input(1)]); in fuse_reorg()1879 shape = get_node_attr_from_input_ai(weights[node3->input(1)]); in fuse_lstm_gru_rnn()4178 pads = get_node_attr_from_input_ai(weights[node.input(1)]); in main()4330 shape = get_node_attr_from_input_ai(weights[node.input(1)]); in main()[all …]