Home
last modified time | relevance | path

Searched refs:TfLiteCustomAllocation (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/c/
Dcommon.h374 typedef struct TfLiteCustomAllocation { struct
377 } TfLiteCustomAllocation; argument
/external/tensorflow/tensorflow/lite/
Dinterpreter_experimental.cc40 int tensor_index, const TfLiteCustomAllocation& allocation, int64_t flags) { in SetCustomAllocationForTensor()
Dinterpreter.h661 int tensor_index, const TfLiteCustomAllocation& allocation,
Dinterpreter_test.cc1568 TfLiteCustomAllocation NewCustomAlloc(size_t num_bytes, in NewCustomAlloc()
1576 return TfLiteCustomAllocation( in NewCustomAlloc()
1600 TfLiteCustomAllocation input_alloc{reinterpret_cast<void*>(dummy_ptr), in TEST_F()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.h356 int tensor_index, const TfLiteCustomAllocation& allocation,
776 std::vector<std::pair<int, TfLiteCustomAllocation>> custom_allocations_;
Dsubgraph.cc1709 int tensor_index, const TfLiteCustomAllocation& allocation, int64_t flags) { in SetCustomAllocationForTensor()
1727 const std::pair<int, TfLiteCustomAllocation>& existing_alloc) { in SetCustomAllocationForTensor()
/external/tensorflow/tensorflow/lite/tools/serialization/
Dwriter_lib_test.cc411 TfLiteCustomAllocation NewCustomAlloc(size_t num_bytes, in NewCustomAlloc()
419 return TfLiteCustomAllocation( in NewCustomAlloc()