Searched refs:nodes_to_partition (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/ |
| D | graph_info_test.cc | 98 const std::vector<int>& nodes_to_partition, in PartitionGraph() argument
101 ConvertVector(nodes_to_partition); in PartitionGraph()
126 std::vector<int> nodes_to_partition = {}; in TEST() local
128 PartitionGraph(graph, nodes_to_partition, &generated_subgraphs); in TEST()
140 std::vector<int> nodes_to_partition = {}; in TEST() local
142 PartitionGraph(graph, nodes_to_partition, &generated_subgraphs); in TEST()
154 std::vector<int> nodes_to_partition = {}; in TEST() local
156 PartitionGraph(graph, nodes_to_partition, &generated_subgraphs); in TEST()
172 std::vector<int> nodes_to_partition = {}; in TEST() local
174 PartitionGraph(graph, nodes_to_partition, &generated_subgraphs); in TEST()
[all …]
|
| D | graph_info.cc | 40 const GraphInfo* info, const TfLiteIntArray* nodes_to_partition, in PartitionGraphIntoIndependentNodeSubsetsImpl() argument
46 for (auto node_index : TfLiteIntArrayView(nodes_to_partition)) { in PartitionGraphIntoIndependentNodeSubsetsImpl()
218 const GraphInfo* info, const TfLiteIntArray* nodes_to_partition, in PartitionGraphIntoIndependentNodeSubsets() argument
220 PartitionGraphIntoIndependentNodeSubsetsImpl(info, nodes_to_partition, in PartitionGraphIntoIndependentNodeSubsets()
|
| D | graph_info.h | 90 const GraphInfo* info, const TfLiteIntArray* nodes_to_partition,
|