Searched refs:TfLiteCustomAllocation (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/lite/c/ |
D | common.h | 374 typedef struct TfLiteCustomAllocation { struct 377 } TfLiteCustomAllocation; argument
|
/external/tensorflow/tensorflow/lite/ |
D | interpreter_experimental.cc | 40 int tensor_index, const TfLiteCustomAllocation& allocation, int64_t flags) { in SetCustomAllocationForTensor()
|
D | interpreter.h | 661 int tensor_index, const TfLiteCustomAllocation& allocation,
|
D | interpreter_test.cc | 1568 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/ |
D | subgraph.h | 356 int tensor_index, const TfLiteCustomAllocation& allocation, 776 std::vector<std::pair<int, TfLiteCustomAllocation>> custom_allocations_;
|
D | subgraph.cc | 1709 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/ |
D | writer_lib_test.cc | 411 TfLiteCustomAllocation NewCustomAlloc(size_t num_bytes, in NewCustomAlloc() 419 return TfLiteCustomAllocation( in NewCustomAlloc()
|