Searched refs:SubgraphAllocations (Results 1 – 9 of 9) sorted by relevance
| /tflite-micro-3.4.0-2.7.6/tensorflow/lite/micro/ |
| D | micro_allocator.h | 83 } SubgraphAllocations; typedef
133 SubgraphAllocations* StartModelAllocation(const Model* model);
148 const Model* model, SubgraphAllocations* subgraph_allocations,
158 const Model* model, const SubgraphAllocations* subgraph_allocations,
169 const Model* model, const SubgraphAllocations* subgraph_allocations,
216 const Model* model, SubgraphAllocations* subgraph_allocations);
222 const Model* model, SubgraphAllocations* subgraph_allocations);
|
| D | micro_graph.h | 74 void SetSubgraphAllocations(SubgraphAllocations* subgraph_allocations);
82 SubgraphAllocations* GetAllocations() { return subgraph_allocations_; } in GetAllocations()
88 SubgraphAllocations* subgraph_allocations_ = nullptr;
|
| D | recording_micro_allocator.h | 75 const Model* model, SubgraphAllocations* subgraph_allocations) override;
77 const Model* model, SubgraphAllocations* subgraph_allocations) override;
|
| D | micro_allocator_test.cc | 86 SubgraphAllocations* subgraph_allocations, int tensor_idx, in VerifyMockTensor()
100 SubgraphAllocations* subgraph_allocations, in VerifyMockWeightTensor()
123 SubgraphAllocations* subgraph_allocations, size_t count, in VerifyRegistrationAndNodeAllocation()
267 tflite::SubgraphAllocations* subgraph_allocations = nullptr; in TF_LITE_MICRO_TEST()
293 tflite::SubgraphAllocations* subgraph_allocations = in TF_LITE_MICRO_TEST()
340 tflite::SubgraphAllocations* subgraph_allocations1 = in TF_LITE_MICRO_TEST()
350 tflite::SubgraphAllocations* subgraph_allocations2 = in TF_LITE_MICRO_TEST()
371 tflite::SubgraphAllocations* subgraph_allocations = in TF_LITE_MICRO_TEST()
415 tflite::SubgraphAllocations* subgraph_allocations = in TF_LITE_MICRO_TEST()
494 tflite::SubgraphAllocations* subgraph_allocations = in TF_LITE_MICRO_TEST()
[all …]
|
| D | micro_allocator.cc | 593 SubgraphAllocations* MicroAllocator::StartModelAllocation(const Model* model) { in StartModelAllocation()
615 SubgraphAllocations* output = reinterpret_cast<SubgraphAllocations*>( in StartModelAllocation()
617 sizeof(SubgraphAllocations) * model->subgraphs()->size(), in StartModelAllocation()
618 alignof(SubgraphAllocations))); in StartModelAllocation()
632 const Model* model, SubgraphAllocations* subgraph_allocations, in FinishModelAllocation()
739 const Model* model, SubgraphAllocations* subgraph_allocations) { in AllocateNodeAndRegistrations()
765 const Model* model, const SubgraphAllocations* subgraph_allocations, in AllocatePersistentTfLiteTensor()
802 const Model* model, const SubgraphAllocations* subgraph_allocations, in AllocateTempTfLiteTensor()
846 const Model* model, SubgraphAllocations* subgraph_allocations) { in AllocateTfLiteEvalTensors()
|
| D | recording_micro_allocator_test.cc | 53 tflite::SubgraphAllocations* subgraph_allocations = in TF_LITE_MICRO_TEST()
92 tflite::SubgraphAllocations* subgraph_allocations = in TF_LITE_MICRO_TEST()
131 tflite::SubgraphAllocations* subgraph_allocations = in TF_LITE_MICRO_TEST()
|
| D | recording_micro_allocator.cc | 133 const Model* model, SubgraphAllocations* subgraph_allocations) { in AllocateNodeAndRegistrations()
158 const Model* model, SubgraphAllocations* subgraph_allocations) { in AllocateTfLiteEvalTensors()
|
| D | micro_graph.cc | 219 SubgraphAllocations* subgraph_allocations) { in SetSubgraphAllocations()
|
| D | micro_interpreter.cc | 184 SubgraphAllocations* allocations = allocator_.StartModelAllocation(model_); in AllocateTensors()
|
