| /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 | 187 RuntimeMemory::RuntimeMemory(SharedMemory memory) : kMemory(std::move(memory)) { in RuntimeMemory() function in android::nn::RuntimeMemory
192 RuntimeMemory::RuntimeMemory(SharedMemory memory, std::unique_ptr<MemoryValidatorBase> validator) in RuntimeMemory() function in android::nn::RuntimeMemory
197 RuntimeMemory::RuntimeMemory(SharedBuffer buffer) : kBuffer(std::move(buffer)) {} in RuntimeMemory() function in android::nn::RuntimeMemory
199 Request::MemoryPool RuntimeMemory::getMemoryPool() const { in getMemoryPool()
206 std::optional<RunTimePoolInfo> RuntimeMemory::getRunTimePoolInfo() const { in getRunTimePoolInfo()
215 void RuntimeMemory::hold(const IBurst::OptionalCacheHold& cacheHold) const { in hold()
273 static int copyInternal(const RuntimeMemory& src, const RuntimeMemory& dst) { in copyInternal()
303 int RuntimeMemory::copy(const RuntimeMemory& src, const RuntimeMemory& dst) { in copy()
484 std::pair<int, std::unique_ptr<RuntimeMemory>> MemoryBuilder::allocate() const { in allocate()
491 std::unique_ptr<RuntimeMemory> memory; in allocate()
[all …]
|
| D | Manager.cpp | 162 std::pair<int, std::unique_ptr<RuntimeMemory>> allocate(const MemoryDescriptor& desc,
196 const std::vector<const RuntimeMemory*>& memories, const SharedBurst& burstController,
204 const std::vector<const RuntimeMemory*>& memories, const std::vector<int>& waitFor,
213 const std::vector<const RuntimeMemory*>& memories, MeasureTiming measure,
239 std::vector<const RuntimeMemory*> memories, MeasureTiming measure, in DriverExecution()
264 const std::vector<const RuntimeMemory*> kMemories;
537 std::pair<int, std::unique_ptr<RuntimeMemory>> DriverDevice::allocate(const MemoryDescriptor& desc, in allocate()
559 const std::vector<const RuntimeMemory*>& memories) { in createDriverRequest()
569 [](const RuntimeMemory* memory) { return memory->getMemoryPool(); }); in createDriverRequest()
581 const std::vector<const RuntimeMemory*>& memories, const SharedBurst& burstController, in execute()
[all …]
|
| D | ExecutionBuilder.h | 51 class RuntimeMemory; variable
69 const RuntimeMemory* memory, size_t offset, size_t length);
73 const RuntimeMemory* memory, size_t offset, size_t length);
391 int setInputFromMemory(uint32_t inputIndex, const RuntimeMemory* memory, uint32_t offset,
396 int setOutputFromMemory(uint32_t outputIndex, const RuntimeMemory* memory, uint32_t offset,
434 int setInputOrOutputFromMemory(const Operand& inputOrOutputOperand, const RuntimeMemory* memory,
|
| D | Manager.h | 82 const std::vector<const RuntimeMemory*>& memories, const SharedBurst& burstController,
93 const std::vector<const RuntimeMemory*>& memories, const std::vector<int>& waitFor,
103 const std::vector<const RuntimeMemory*>& memories, MeasureTiming measure,
162 virtual std::pair<int, std::unique_ptr<RuntimeMemory>> allocate(const MemoryDescriptor& desc,
|
| D | NeuralNetworks.cpp | 1056 const RuntimeMemory* s = reinterpret_cast<const RuntimeMemory*>(src); in ANeuralNetworksMemory_copy()
1057 const RuntimeMemory* d = reinterpret_cast<const RuntimeMemory*>(dst); in ANeuralNetworksMemory_copy()
1058 return RuntimeMemory::copy(*s, *d); in ANeuralNetworksMemory_copy()
1104 RuntimeMemory* m = reinterpret_cast<RuntimeMemory*>(memory); in ANeuralNetworksMemory_free()
1171 const RuntimeMemory* mem = reinterpret_cast<const RuntimeMemory*>(memory); in ANeuralNetworksModel_setOperandValueFromMemory()
1388 const RuntimeMemory* m = reinterpret_cast<const RuntimeMemory*>(memory); in ANeuralNetworksExecution_setInputFromMemory()
1416 const RuntimeMemory* m = reinterpret_cast<const RuntimeMemory*>(memory); in ANeuralNetworksExecution_setOutputFromMemory()
|
| D | ExecutionBuilder.cpp | 213 const RuntimeMemory* memory, size_t offset, in setInputFromMemory()
294 const RuntimeMemory* memory, size_t offset, in setOutputFromMemory()
1071 const RuntimeMemory* memory = mMemories[p.locationAndLength().poolIndex]; in compute()
1191 const RuntimeMemory* memory = mMemories[output.locationAndLength().poolIndex]; in updateMemories()
1206 const RuntimeMemory* memory = mMemories[output.locationAndLength().poolIndex]; in finishComputation()
1414 const RuntimeMemory* memory = mExecutionBuilder->mMemories[builderPoolIndex]; in mapInputOrOutput()
1423 const RuntimeMemory* memory, uint32_t offset, in setInputOrOutputFromMemory()
1595 std::vector<const RuntimeMemory*> memories = mMemories.getObjects(); in computeOnCpuFallback()
1598 std::vector<std::unique_ptr<RuntimeMemory>> blobAhwbs; in computeOnCpuFallback()
1622 const RuntimeMemory* memory = mMemories[i]; in computeOnCpuFallback()
[all …]
|
| D | NeuralNetworksV2.cpp | 1011 const RuntimeMemory* s = reinterpret_cast<const RuntimeMemory*>(src); in ANeuralNetworksMemory_copy()
1012 const RuntimeMemory* d = reinterpret_cast<const RuntimeMemory*>(dst); in ANeuralNetworksMemory_copy()
1013 return RuntimeMemory::copy(*s, *d); in ANeuralNetworksMemory_copy()
1059 RuntimeMemory* m = reinterpret_cast<RuntimeMemory*>(memory); in ANeuralNetworksMemory_free()
1126 const RuntimeMemory* mem = reinterpret_cast<const RuntimeMemory*>(memory); in ANeuralNetworksModel_setOperandValueFromMemory()
|
| D | ModelBuilder.h | 38 class RuntimeMemory; variable
48 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 | 324 int ModelBuilder::setOperandValueFromMemory(uint32_t index, const RuntimeMemory* memory, in setOperandValueFromMemory()
941 [](const RuntimeMemory* m) { return m->getMemory(); }); in makeModel()
|
| D | ExecutionPlan.cpp | 499 const RuntimeMemory* memory = sourceModel.getMemories()[operand.location.poolIndex]; in addOperand()
591 const std::vector<OutputShape>* mainModelOutputShapes, const RuntimeMemory* temporaryMemory, in mapInputsAndOutputs()
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestMemoryDomain.cpp | 305 const RuntimeMemory* m = reinterpret_cast<const RuntimeMemory*>(memory.get()); in TEST_P()
315 const RuntimeMemory* m = reinterpret_cast<const RuntimeMemory*>(memory.get()); in TEST_P()
343 const RuntimeMemory* m = reinterpret_cast<const RuntimeMemory*>(memory.get()); in TEST_P()
353 const RuntimeMemory* m = reinterpret_cast<const RuntimeMemory*>(memory.get()); in TEST_P()
377 const RuntimeMemory* m = reinterpret_cast<const RuntimeMemory*>(memory.get()); in TEST_P()
400 const RuntimeMemory* m = reinterpret_cast<const RuntimeMemory*>(memory.get()); in TEST_P()
427 const RuntimeMemory* m = reinterpret_cast<const RuntimeMemory*>(memory.get()); in TEST_P()
|