Searched refs:GraphPartitionHelper (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/ |
| D | utils.h | 50 class GraphPartitionHelper {
52 GraphPartitionHelper(TfLiteContext* context, in GraphPartitionHelper() function
56 GraphPartitionHelper(TfLiteContext* context, in GraphPartitionHelper() function
63 virtual ~GraphPartitionHelper() { TfLiteIntArrayFree(supported_nodes_); } in ~GraphPartitionHelper()
140 class FP16GraphPartitionHelper : public GraphPartitionHelper {
144 : GraphPartitionHelper(context, std::move(is_node_supported_fn)) {} in FP16GraphPartitionHelper()
|
| D | utils_test.cc | 186 TEST(GraphPartitionHelper, CheckPartitions) { in TEST() argument
189 GraphPartitionHelper helper(&mocked_context, IsNodeSupported); in TEST()
218 TEST(GraphPartitionHelper, CheckPrepareErrors) { in TEST() argument
222 GraphPartitionHelper helper(&error_context, IsNodeSupported); in TEST()
226 TEST(GraphPartitionHelper, CheckPartitionsWithSupportedNodeList) { in TEST() argument
231 GraphPartitionHelper helper(&mocked_context, supported_nodes); in TEST()
|
| D | utils.cc | 50 TfLiteStatus GraphPartitionHelper::Partition( in Partition()
72 GraphPartitionHelper::GetFirstNLargestPartitions( in GetFirstNLargestPartitions()
99 std::vector<int> GraphPartitionHelper::GetNodesOfFirstNLargestPartitionsImpl( in GetNodesOfFirstNLargestPartitionsImpl()
112 TfLiteStatus GraphPartitionHelper::PrepareSupportedNodes( in PrepareSupportedNodes()
249 const auto is_supported = GraphPartitionHelper::IsNodeSupported( in IsNodeSupported()
|
| /external/tensorflow/tensorflow/lite/delegates/utils/ |
| D | simple_delegate.cc | 94 delegates::GraphPartitionHelper helper(context, node_supported_fn); in DelegatePrepare()
|