Searched refs:TfLiteCustomAllocation (Results 1 – 7 of 7) sorted by relevance
373 typedef struct TfLiteCustomAllocation { struct376 } TfLiteCustomAllocation; argument
348 int tensor_index, const TfLiteCustomAllocation& allocation);713 std::vector<std::pair<int, TfLiteCustomAllocation>> custom_allocations_;
160 const TfLiteCustomAllocation& allocation) { in ValidateCustomAllocationForTensor()1594 int tensor_index, const TfLiteCustomAllocation& allocation) { in SetCustomAllocationForTensor()1607 const std::pair<int, TfLiteCustomAllocation>& existing_alloc) { in SetCustomAllocationForTensor()
409 TfLiteCustomAllocation NewCustomAlloc(size_t num_bytes, in NewCustomAlloc()417 return TfLiteCustomAllocation( in NewCustomAlloc()
164 int tensor_index, const TfLiteCustomAllocation& allocation) { in SetCustomAllocationForTensor()
622 int tensor_index, const TfLiteCustomAllocation& allocation);
1565 TfLiteCustomAllocation NewCustomAlloc(size_t num_bytes, in NewCustomAlloc()1573 return TfLiteCustomAllocation( in NewCustomAlloc()1597 TfLiteCustomAllocation input_alloc{reinterpret_cast<void*>(dummy_ptr), in TEST_F()