Searched refs:node_subset (Results 1 – 2 of 2) sorted by relevance
309 const NodeSubset& node_subset) { in CreateDelegateParams() argument314 TfLiteIntArrayGetSizeInBytes(node_subset.nodes.size()); in CreateDelegateParams()318 TfLiteIntArrayGetSizeInBytes(node_subset.input_tensors.size()); in CreateDelegateParams()322 TfLiteIntArrayGetSizeInBytes(node_subset.output_tensors.size()); in CreateDelegateParams()336 CopyVectorToTfLiteIntArray(node_subset.nodes, params->nodes_to_replace); in CreateDelegateParams()340 CopyVectorToTfLiteIntArray(node_subset.input_tensors, params->input_tensors); in CreateDelegateParams()344 CopyVectorToTfLiteIntArray(node_subset.output_tensors, in CreateDelegateParams()352 void PopulatePreviewDelegateParams(const NodeSubset& node_subset, in PopulatePreviewDelegateParams() argument358 params->nodes_to_replace = TfLiteIntArrayCreate(node_subset.nodes.size()); in PopulatePreviewDelegateParams()359 CopyVectorToTfLiteIntArray(node_subset.nodes, params->nodes_to_replace); in PopulatePreviewDelegateParams()[all …]
94 for (NodeSubset& node_subset : *node_subsets_) { in Partition()101 uniquefy(&node_subset.input_tensors); in Partition()102 uniquefy(&node_subset.output_tensors); in Partition()