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

/frameworks/ml/nn/runtime/
H A DMemory.h98 const Memory* operator[](size_t i) const { return mMemories[i]; }
102 std::vector<const Memory*> mMemories; member in class:android::nn::MemoryTracker
H A DExecutionBuilder.cpp139 uint32_t poolIndex = mMemories.add(memory);
179 uint32_t poolIndex = mMemories.add(memory);
386 uint32_t nextPoolIndex = mMemories.size();
406 mMemories.add(memory);
434 mMemories = mExecutionBuilder->mMemories;
449 const Memory* memory = mExecutionBuilder->mMemories[builderPoolIndex];
450 const uint32_t executorPoolIndex = mMemories.add(memory);
465 uint32_t poolIndex = mMemories.add(memory);
540 uint32_t count = mMemories
[all...]
H A DModelBuilder.h78 const MemoryTracker& getMemories() const { return mMemories; }
116 MemoryTracker mMemories; member in class:android::nn::ModelBuilder
H A DExecutionBuilder.h98 MemoryTracker mMemories; member in class:android::nn::ExecutionBuilder
191 MemoryTracker mMemories; member in class:android::nn::StepExecutor
H A DMemory.cpp128 mMemories.push_back(memory);
H A DModelBuilder.cpp138 uint32_t poolIndex = mMemories.add(&mLargeValueMemory);
170 .poolIndex = mMemories.add(memory), .offset = offset, .length = neededLength};
350 uint32_t count = mMemories.size();
353 model->pools[i] = mMemories[i]->getHidlMemory();

Completed in 155 milliseconds