• Home
  • Raw
  • Download

Lines Matching refs:outputIndex

1553 +  if ((in_nums > 0 && node->inputIndex.size() != in_nums) || node->outputIndex.size() != out_nums)…
1558 + std::all_of(node->outputIndex.begin(), node->outputIndex.end(),
1574 + (void)tensor_indexes.insert(tensor_indexes.end(), node->outputIndex.begin(), node->outputIndex.e…
1632 + auto out_shape = graph->allTensors[node->outputIndex.front()]->dims;
1641 + if (node->inputIndex.size() != 1 || node->outputIndex.size() != kSplitOutSize) {
1645 + auto out_shape0 = graph->allTensors[node->outputIndex[0]]->dims;
1646 + auto out_shape1 = graph->allTensors[node->outputIndex[1]]->dims;
1647 + auto out_shape2 = graph->allTensors[node->outputIndex[kInputSize1]]->dims;
1667 + auto out_shape = graph->allTensors[node->outputIndex.front()]->dims;
1740 + auto out_shape = graph->allTensors[node->outputIndex.back()]->dims;
1831 + auto out_indexes = all_nodes[node_index]->outputIndex;
1866 + UpdateTensorIndex(&node->outputIndex);
1868 + UpdateTensorIndex(&graph_->outputIndex);
1870 + main_graph->outputIndices = graph_->outputIndex;
1897 + InsertElements(node->outputIndex);
1899 + InsertElements(graph_->outputIndex);
2002 + auto out_tensor_indexes = graph_->nodes[node_index]->outputIndex;
2074 + auto out_tensor_indexes = node->outputIndex;
2124 + auto outputs = graph_->nodes[real_node_map_[kAdd0]]->outputIndex;
2135 + custom_gru->outputIndex = outputs;
2260 + auto init_h = first_node->outputIndex.front();
2274 + init_h = node->outputIndex.front();
2284 + gru_cell_node->outputIndex[0] = graph->nodes[stack_index]->outputIndex[0];