Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/runtime/
H A DExecutionBuilder.cpp244 uint32_t poolIndex = mMemories.add(memory); in setInputFromMemory()
325 uint32_t poolIndex = mMemories.add(memory); in setOutputFromMemory()
1027 const RuntimeMemory* memory = mMemories[p.locationAndLength().poolIndex]; in compute()
1146 const RuntimeMemory* memory = mMemories[output.locationAndLength().poolIndex]; in updateMemories()
1340 mMemories = mExecutionBuilder->mMemories; in mapInputsAndOutputsTrivially()
1368 const uint32_t executorPoolIndex = mMemories.add(memory); in mapInputOrOutput()
1383 uint32_t poolIndex = mMemories.add(memory); in setInputOrOutputFromMemory()
1546 std::vector<const RuntimeMemory*> memories = mMemories.getObjects(); in computeOnCpuFallback()
1562 if (mMemories[poolIndex]->getValidator().createdWithUnknownShape()) { in computeOnCpuFallback()
1573 const RuntimeMemory* memory = mMemories[i]; in computeOnCpuFallback()
[all …]
H A DModelBuilder.cpp304 uint32_t poolIndex = mMemories.add(mLargeValueMemory.get()); in copyLargeValuesToSharedMemory()
350 operand.location = {.poolIndex = mMemories.add(memory), in setOperandValueFromMemory()
882 MemoryTracker mMemories; member in android::nn::ModelBuilder::ModelMaker
904 model.pools.reserve(mMemories.size()); in makeModel()
905 std::transform(mMemories.begin(), mMemories.end(), std::back_inserter(model.pools), in makeModel()
931 operand.location.poolIndex = mMemories.add(refModel->mMemories[originalPoolIndex]); in updateOperandLocations()
H A DExecutionBuilder.h137 return mMemories[poolIndex]->getRunTimePoolInfo(); in getRunTimePoolInfo()
188 MemoryTracker mMemories; variable
449 MemoryTracker mMemories; variable
H A DModelBuilder.h109 const MemoryTracker& getMemories() const { return mMemories; } in getMemories()
202 MemoryTracker mMemories; variable
/aosp12/packages/modules/NeuralNetworks/runtime/test/
H A DTestPartitioningRandom.cpp297 const WrapperMemory* memory = &mMemories[std::get<0>(regionDescriptor)]; in getRegion()
321 std::vector<WrapperMemory> mMemories; member in android::__anonc614c5800110::TestMemories
339 mMemories.emplace_back(memory); in layout()
H A DTestValidation.cpp245 for (auto* memory : mMemories) ANeuralNetworksMemory_free(memory); in TearDown()
257 mMemories.push_back(ashmem); in createAshmem()
262 std::vector<ANeuralNetworksMemory*> mMemories; member in __anonb91444320110::ValidationTestMemoryDesc