Searched refs:safe_allocator (Results 1 – 2 of 2) sorted by relevance
161 SafeBuiltinDataAllocator safe_allocator(allocator); in ParseOpDataTfLite() local481 auto params = safe_allocator.Allocate<TfLiteLSHProjectionParams>(); in ParseOpDataTfLite()491 auto params = safe_allocator.Allocate<TfLiteSequenceRNNParams>(); in ParseOpDataTfLite()506 safe_allocator.Allocate<TfLiteBidirectionalSequenceRNNParams>(); in ParseOpDataTfLite()521 auto params = safe_allocator.Allocate<TfLiteRNNParams>(); in ParseOpDataTfLite()534 safe_allocator.Allocate<TfLiteEmbeddingLookupSparseParams>(); in ParseOpDataTfLite()549 auto params = safe_allocator.Allocate<TfLiteLocalResponseNormParams>(); in ParseOpDataTfLite()562 auto params = safe_allocator.Allocate<TfLiteLSTMParams>(); in ParseOpDataTfLite()594 safe_allocator.Allocate<TfLiteUnidirectionalSequenceLSTMParams>(); in ParseOpDataTfLite()611 safe_allocator.Allocate<TfLiteBidirectionalSequenceLSTMParams>(); in ParseOpDataTfLite()[all …]
122 SafeBuiltinDataAllocator safe_allocator(allocator);126 params = safe_allocator.Allocate<TfLiteDepthwiseConvParams>();