Searched refs:tensor_idx_to_alloc (Results 1 – 1 of 1) sorted by relevance
176 const std::map<int, TfLiteCustomAllocation>& tensor_idx_to_alloc, in VerifyCustomAllocationForTensor() argument180 const auto idx_and_alloc = tensor_idx_to_alloc.find(tensor_idx); in VerifyCustomAllocationForTensor()181 TF_LITE_ENSURE(context, idx_and_alloc != tensor_idx_to_alloc.end()); in VerifyCustomAllocationForTensor()