Searched refs:node_subset (Results 1 – 2 of 2) sorted by relevance
226 const NodeSubset& node_subset) { in CreateDelegateParams() argument231 TfLiteIntArrayGetSizeInBytes(node_subset.nodes.size()); in CreateDelegateParams()235 TfLiteIntArrayGetSizeInBytes(node_subset.input_tensors.size()); in CreateDelegateParams()239 TfLiteIntArrayGetSizeInBytes(node_subset.output_tensors.size()); in CreateDelegateParams()253 CopyVectorToTfLiteIntArray(node_subset.nodes, params->nodes_to_replace); in CreateDelegateParams()257 CopyVectorToTfLiteIntArray(node_subset.input_tensors, params->input_tensors); in CreateDelegateParams()261 CopyVectorToTfLiteIntArray(node_subset.output_tensors, in CreateDelegateParams()285 for (auto& node_subset : node_subsets) { in ReplaceNodeSubsetsWithDelegateKernels() local289 switch (node_subset.type) { in ReplaceNodeSubsetsWithDelegateKernels()291 for (auto it = node_subset.nodes.begin(); it != node_subset.nodes.end(); in ReplaceNodeSubsetsWithDelegateKernels()[all …]
108 for (NodeSubset& node_subset : *node_subsets_) { in Partition()115 uniquefy(&node_subset.input_tensors); in Partition()116 uniquefy(&node_subset.output_tensors); in Partition()