Home
last modified time | relevance | path

Searched refs:mMemories (Results 1 – 6 of 6) sorted by relevance

/frameworks/ml/nn/runtime/
DMemory.h180 std::vector<const Memory*> mMemories;
191 const Memory* operator[](size_t i) const { return mMemories[i]; }
193 decltype(mMemories.begin()) begin() { return mMemories.begin(); } in begin()
194 decltype(mMemories.end()) end() { return mMemories.end(); } in end()
DExecutionBuilder.cpp225 uint32_t poolIndex = mMemories.add(memory); in setInputFromMemory()
282 uint32_t poolIndex = mMemories.add(memory); in setOutputFromMemory()
677 uint32_t nextPoolIndex = mMemories.size(); in allocatePointerArgumentsToPool()
697 mMemories.add(memory); in allocatePointerArgumentsToPool()
731 mMemories = mExecutionBuilder->mMemories; in mapInputsAndOutputsTrivially()
747 const Memory* memory = mExecutionBuilder->mMemories[builderPoolIndex]; in mapInputOrOutput()
748 const uint32_t executorPoolIndex = mMemories.add(memory); in mapInputOrOutput()
762 uint32_t poolIndex = mMemories.add(memory); in setInputOrOutputFromTemporaryMemory()
880 uint32_t count = mMemories.size(); in startComputeOnDevice()
883 request.pools[i] = mMemories[i]->getHidlMemory(); in startComputeOnDevice()
[all …]
DModelBuilder.h93 const MemoryTracker& getMemories() const { return mMemories; } in getMemories()
148 MemoryTracker mMemories; variable
DExecutionBuilder.h148 MemoryTracker mMemories; variable
270 MemoryTracker mMemories; variable
DModelBuilder.cpp264 uint32_t poolIndex = mMemories.add(&mLargeValueMemory); in copyLargeValuesToSharedMemory()
313 operand.location = {.poolIndex = mMemories.add(memory), in setOperandValueFromMemory()
539 uint32_t count = mMemories.size(); in setHidlModel()
542 model->pools[i] = mMemories[i]->getHidlMemory(); in setHidlModel()
DMemory.cpp158 mMemories.push_back(memory); in add()