Searched refs:nodes_to_partition (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/ |
| D | graph_info_test.cc | 78 const std::vector<int>& nodes_to_partition, in PartitionGraph() argument
81 ConvertVector(nodes_to_partition); in PartitionGraph()
106 std::vector<int> nodes_to_partition = {}; in TEST() local
108 PartitionGraph(graph, nodes_to_partition, &generated_subgraphs); in TEST()
120 std::vector<int> nodes_to_partition = {}; in TEST() local
122 PartitionGraph(graph, nodes_to_partition, &generated_subgraphs); in TEST()
134 std::vector<int> nodes_to_partition = {}; in TEST() local
136 PartitionGraph(graph, nodes_to_partition, &generated_subgraphs); in TEST()
155 std::vector<int> nodes_to_partition = {0}; in TEST() local
156 PartitionGraph(graph, nodes_to_partition, &generated_subgraphs); in TEST()
[all …]
|
| D | graph_info.cc | 55 const GraphInfo* info, const TfLiteIntArray* nodes_to_partition, in PartitionGraphIntoIndependentNodeSubsetsImpl() argument
61 for (auto node_index : TfLiteIntArrayView(nodes_to_partition)) { in PartitionGraphIntoIndependentNodeSubsetsImpl()
225 const GraphInfo* info, const TfLiteIntArray* nodes_to_partition, in PartitionGraphIntoIndependentNodeSubsets() argument
227 PartitionGraphIntoIndependentNodeSubsetsImpl(info, nodes_to_partition, in PartitionGraphIntoIndependentNodeSubsets()
|
| D | graph_info.h | 78 const GraphInfo* info, const TfLiteIntArray* nodes_to_partition,
|