Searched refs:GraphPartitionHelper (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/ |
| H A D | utils.h | 50 class GraphPartitionHelper {
52 GraphPartitionHelper(TfLiteContext* context, in GraphPartitionHelper() function
56 GraphPartitionHelper(TfLiteContext* context, in GraphPartitionHelper() function
63 virtual ~GraphPartitionHelper() { in ~GraphPartitionHelper()
150 class FP16GraphPartitionHelper : public GraphPartitionHelper {
154 : GraphPartitionHelper(context, std::move(is_node_supported_fn)) {} in FP16GraphPartitionHelper()
|
| H A D | utils_test.cc | 187 TEST(GraphPartitionHelper, CheckPartitions) { in TEST() argument
190 GraphPartitionHelper helper(&mocked_context, IsNodeSupported); in TEST()
219 TEST(GraphPartitionHelper, CheckPrepareErrors) { in TEST() argument
223 GraphPartitionHelper helper(&error_context, IsNodeSupported); in TEST()
227 TEST(GraphPartitionHelper, CheckPartitionsWithSupportedNodeList) { in TEST() argument
232 GraphPartitionHelper helper(&mocked_context, supported_nodes); in TEST()
|
| H A D | utils.cc | 53 TfLiteStatus GraphPartitionHelper::Partition( in Partition()
75 GraphPartitionHelper::GetFirstNLargestPartitions( in GetFirstNLargestPartitions()
102 std::vector<int> GraphPartitionHelper::GetNodesOfFirstNLargestPartitionsImpl( in GetNodesOfFirstNLargestPartitionsImpl()
115 TfLiteStatus GraphPartitionHelper::PrepareSupportedNodes( in PrepareSupportedNodes()
230 const auto is_supported = GraphPartitionHelper::IsNodeSupported( in IsNodeSupported()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/utils/ |
| H A D | simple_delegate.cc | 95 delegates::GraphPartitionHelper helper(context, node_supported_fn); in DelegatePrepare()
|