Searched refs:partition_helper (Results 1 – 3 of 3) sorted by relevance
243 delegates::FP16GraphPartitionHelper partition_helper(context, node_supported_fn);244 TF_LITE_ENSURE_STATUS(partition_helper.Partition(nullptr));246 std::vector<int> delegated_nodes = partition_helper.GetNodesOfFirstNLargestPartitions(251 delegated_nodes.size(), partition_helper.num_total_nodes(),252 partition_helper.num_partitions());
435 delegates::FP16GraphPartitionHelper partition_helper(context, in FP16Delegate() local438 if (partition_helper.Partition(nullptr) != kTfLiteOk) { in FP16Delegate()442 partition_helper.GetNodesOfFirstNLargestPartitions( in FP16Delegate()
2504 delegates::FP16GraphPartitionHelper partition_helper(context, in GetOpsToReplace() local2507 if (partition_helper.Partition(&unsupported_nodes_info) != kTfLiteOk) { in GetOpsToReplace()2514 partition_helper.GetNodesOfFirstNLargestPartitions( in GetOpsToReplace()2526 partition_helper.num_total_nodes() - ops_to_replace.size()); in GetOpsToReplace()2530 partition_helper.num_total_nodes()); in GetOpsToReplace()