| /aosp12/packages/modules/NeuralNetworks/runtime/ |
| H A 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);
200 explicit RuntimeMemory(SharedBuffer buffer);
264 class MemoryAshmem : public RuntimeMemory {
291 class MemoryFd : public RuntimeMemory {
305 class MemoryAHWB : public RuntimeMemory {
319 class MemoryRuntimeAHWB : public RuntimeMemory {
[all …]
|
| H A 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()
479 std::unique_ptr<RuntimeMemory> memory; in allocate()
528 : RuntimeMemory(std::move(memory)), kMapping(std::move(mapping)) {} in MemoryAshmem()
[all …]
|
| H A D | Manager.cpp | 118 std::pair<int, std::unique_ptr<RuntimeMemory>> allocate(const MemoryDescriptor& desc,
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;
540 for (const RuntimeMemory* memory : memories) { in execute()
651 const std::vector<const RuntimeMemory*>& memories, MeasureTiming measure, in createReusableExecution()
679 for (const RuntimeMemory* memory : kMemories) { in compute()
918 const std::vector<const RuntimeMemory*>& memories, MeasureTiming measure,
1078 const std::vector<const RuntimeMemory*>& memories) { in createCpuRequest()
1081 for (const RuntimeMemory* mem : memories) { in createCpuRequest()
[all …]
|
| H A 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,
|
| H A 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,
|
| H A 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 …]
|
| H A 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()
|
| H A D | ModelBuilder.h | 37 class RuntimeMemory; variable
47 int setOperandValueFromMemory(uint32_t index, const RuntimeMemory* memory, uint32_t offset,
|
| H A D | ExecutionPlan.h | 55 class RuntimeMemory; variable
185 const RuntimeMemory* memory;
298 const RuntimeMemory* temporaryMemory, // for static temporaries
556 const RuntimeMemory* memory;
|
| H A D | ModelBuilder.cpp | 318 int ModelBuilder::setOperandValueFromMemory(uint32_t index, const RuntimeMemory* memory, in setOperandValueFromMemory()
906 [](const RuntimeMemory* m) { return m->getMemory(); }); in makeModel()
|
| H A 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()
|
| /aosp12/packages/modules/NeuralNetworks/runtime/test/ |
| H A 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()
|