Searched refs:subgraph_index (Results 1 – 3 of 3) sorted by relevance
92 for (int subgraph_index = 0; subgraph_index < generated_subgraphs.size(); in CheckPartitionSubgraphs() local93 subgraph_index++) { in CheckPartitionSubgraphs()94 EXPECT_EQ(generated_subgraphs[subgraph_index].nodes, in CheckPartitionSubgraphs()95 expected_subgraphs[subgraph_index].nodes); in CheckPartitionSubgraphs()96 EXPECT_EQ(generated_subgraphs[subgraph_index].input_tensors, in CheckPartitionSubgraphs()97 expected_subgraphs[subgraph_index].input_tensors); in CheckPartitionSubgraphs()98 EXPECT_EQ(generated_subgraphs[subgraph_index].output_tensors, in CheckPartitionSubgraphs()99 expected_subgraphs[subgraph_index].output_tensors); in CheckPartitionSubgraphs()
461 Subgraph* subgraph(int subgraph_index) { in subgraph() argument462 if (subgraph_index < 0 || in subgraph()463 static_cast<size_t>(subgraph_index) >= subgraphs_size()) in subgraph()465 return &*subgraphs_[subgraph_index]; in subgraph()
524 for (int subgraph_index = 0; subgraph_index < subgraphs->Length(); in operator ()() local525 ++subgraph_index) { in operator ()()526 const tflite::SubGraph* subgraph = (*subgraphs)[subgraph_index]; in operator ()()528 (*interpreter)->subgraph(subgraph_index); in operator ()()534 subgraph_index); in operator ()()