| /packages/modules/NeuralNetworks/runtime/ |
| D | Memory.h | 168 class RuntimeMemory {
170 DISALLOW_COPY_AND_ASSIGN(RuntimeMemory);
173 virtual ~RuntimeMemory() = default;
195 static int copy(const RuntimeMemory& src, const RuntimeMemory& dst);
198 explicit RuntimeMemory(SharedMemory memory);
199 RuntimeMemory(SharedMemory memory, std::unique_ptr<MemoryValidatorBase> validator);
200 explicit RuntimeMemory(SharedBuffer buffer);
231 std::pair<int, std::unique_ptr<RuntimeMemory>> allocate() const;
264 class MemoryAshmem : public RuntimeMemory {
291 class MemoryFd : public RuntimeMemory {
[all …]
|
| D | Memory.cpp | 183 RuntimeMemory::RuntimeMemory(SharedMemory memory) : kMemory(std::move(memory)) { in RuntimeMemory() function in android::nn::RuntimeMemory
188 RuntimeMemory::RuntimeMemory(SharedMemory memory, std::unique_ptr<MemoryValidatorBase> validator) in RuntimeMemory() function in android::nn::RuntimeMemory
193 RuntimeMemory::RuntimeMemory(SharedBuffer buffer) : kBuffer(std::move(buffer)) {} in RuntimeMemory() function in android::nn::RuntimeMemory
195 Request::MemoryPool RuntimeMemory::getMemoryPool() const { in getMemoryPool()
202 std::optional<RunTimePoolInfo> RuntimeMemory::getRunTimePoolInfo() const { in getRunTimePoolInfo()
211 void RuntimeMemory::hold(const IBurst::OptionalCacheHold& cacheHold) const { in hold()
265 static int copyInternal(const RuntimeMemory& src, const RuntimeMemory& dst) { in copyInternal()
295 int RuntimeMemory::copy(const RuntimeMemory& src, const RuntimeMemory& dst) { in copy()
472 std::pair<int, std::unique_ptr<RuntimeMemory>> MemoryBuilder::allocate() const { in allocate()
479 std::unique_ptr<RuntimeMemory> memory; in allocate()
[all …]
|
| D | Manager.cpp | 118 std::pair<int, std::unique_ptr<RuntimeMemory>> allocate(const MemoryDescriptor& desc,
153 const std::vector<const RuntimeMemory*>& memories, const SharedBurst& burstController,
160 const std::vector<const RuntimeMemory*>& memories, const std::vector<int>& waitFor,
168 const std::vector<const RuntimeMemory*>& memories, MeasureTiming measure,
193 std::vector<const RuntimeMemory*> memories, MeasureTiming measure, in DriverExecution()
216 const std::vector<const RuntimeMemory*> kMemories;
482 std::pair<int, std::unique_ptr<RuntimeMemory>> DriverDevice::allocate(const MemoryDescriptor& desc, in allocate()
504 const std::vector<const RuntimeMemory*>& memories) { in createDriverRequest()
514 [](const RuntimeMemory* memory) { return memory->getMemoryPool(); }); in createDriverRequest()
526 const std::vector<const RuntimeMemory*>& memories, const SharedBurst& burstController, in execute()
[all …]
|
| D | ExecutionBuilder.h | 50 class RuntimeMemory; variable
65 const RuntimeMemory* memory, size_t offset, size_t length);
69 const RuntimeMemory* memory, size_t offset, size_t length);
369 int setInputFromMemory(uint32_t inputIndex, const RuntimeMemory* memory, uint32_t offset,
374 int setOutputFromMemory(uint32_t outputIndex, const RuntimeMemory* memory, uint32_t offset,
412 int setInputOrOutputFromMemory(const Operand& inputOrOutputOperand, const RuntimeMemory* memory,
|
| D | Manager.h | 81 const std::vector<const RuntimeMemory*>& memories, const SharedBurst& burstController,
91 const std::vector<const RuntimeMemory*>& memories, const std::vector<int>& waitFor,
100 const std::vector<const RuntimeMemory*>& memories, MeasureTiming measure,
157 virtual std::pair<int, std::unique_ptr<RuntimeMemory>> allocate(const MemoryDescriptor& desc,
|
| D | ExecutionBuilder.cpp | 212 const RuntimeMemory* memory, size_t offset, in setInputFromMemory()
293 const RuntimeMemory* memory, size_t offset, in setOutputFromMemory()
1027 const RuntimeMemory* memory = mMemories[p.locationAndLength().poolIndex]; in compute()
1146 const RuntimeMemory* memory = mMemories[output.locationAndLength().poolIndex]; in updateMemories()
1160 const RuntimeMemory* memory = mMemories[output.locationAndLength().poolIndex]; in finishComputation()
1367 const RuntimeMemory* memory = mExecutionBuilder->mMemories[builderPoolIndex]; in mapInputOrOutput()
1376 const RuntimeMemory* memory, uint32_t offset, in setInputOrOutputFromMemory()
1546 std::vector<const RuntimeMemory*> memories = mMemories.getObjects(); in computeOnCpuFallback()
1549 std::vector<std::unique_ptr<RuntimeMemory>> blobAhwbs; in computeOnCpuFallback()
1573 const RuntimeMemory* memory = mMemories[i]; in computeOnCpuFallback()
[all …]
|
| D | NeuralNetworks.cpp | 1016 const RuntimeMemory* s = reinterpret_cast<const RuntimeMemory*>(src); in ANeuralNetworksMemory_copy()
1017 const RuntimeMemory* d = reinterpret_cast<const RuntimeMemory*>(dst); in ANeuralNetworksMemory_copy()
1018 return RuntimeMemory::copy(*s, *d); in ANeuralNetworksMemory_copy()
1052 RuntimeMemory* m = reinterpret_cast<RuntimeMemory*>(memory); in ANeuralNetworksMemory_free()
1119 const RuntimeMemory* mem = reinterpret_cast<const RuntimeMemory*>(memory); in ANeuralNetworksModel_setOperandValueFromMemory()
1337 const RuntimeMemory* m = reinterpret_cast<const RuntimeMemory*>(memory); in ANeuralNetworksExecution_setInputFromMemory()
1365 const RuntimeMemory* m = reinterpret_cast<const RuntimeMemory*>(memory); in ANeuralNetworksExecution_setOutputFromMemory()
|
| D | ModelBuilder.h | 37 class RuntimeMemory; variable
47 int setOperandValueFromMemory(uint32_t index, const RuntimeMemory* memory, uint32_t offset,
|
| D | ExecutionPlan.h | 55 class RuntimeMemory; variable
185 const RuntimeMemory* memory;
298 const RuntimeMemory* temporaryMemory, // for static temporaries
556 const RuntimeMemory* memory;
|
| D | ModelBuilder.cpp | 318 int ModelBuilder::setOperandValueFromMemory(uint32_t index, const RuntimeMemory* memory, in setOperandValueFromMemory()
906 [](const RuntimeMemory* m) { return m->getMemory(); }); in makeModel()
|
| D | ExecutionPlan.cpp | 459 const RuntimeMemory* memory = sourceModel.getMemories()[operand.location.poolIndex]; in addOperand()
551 const std::vector<OutputShape>* mainModelOutputShapes, const RuntimeMemory* temporaryMemory, in mapInputsAndOutputs()
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestMemoryDomain.cpp | 302 const RuntimeMemory* m = reinterpret_cast<const RuntimeMemory*>(memory.get()); in TEST_P()
312 const RuntimeMemory* m = reinterpret_cast<const RuntimeMemory*>(memory.get()); in TEST_P()
340 const RuntimeMemory* m = reinterpret_cast<const RuntimeMemory*>(memory.get()); in TEST_P()
350 const RuntimeMemory* m = reinterpret_cast<const RuntimeMemory*>(memory.get()); in TEST_P()
374 const RuntimeMemory* m = reinterpret_cast<const RuntimeMemory*>(memory.get()); in TEST_P()
397 const RuntimeMemory* m = reinterpret_cast<const RuntimeMemory*>(memory.get()); in TEST_P()
424 const RuntimeMemory* m = reinterpret_cast<const RuntimeMemory*>(memory.get()); in TEST_P()
|