Home
last modified time | relevance | path

Searched defs:poolIndex (Results 1 – 7 of 7) sorted by relevance

/frameworks/ml/nn/runtime/
DExecutionBuilder.cpp193 uint32_t poolIndex = mMemories.add(memory); in setInputFromMemory() local
270 uint32_t poolIndex = mMemories.add(memory); in setOutputFromMemory() local
1027 uint32_t poolIndex = mMemories.add(memory); in setInputOrOutputFromMemory() local
1153 const uint32_t poolIndex = input.locationAndLength().poolIndex; in computeOnCpuFallback() local
1159 const uint32_t poolIndex = output.locationAndLength().poolIndex; in computeOnCpuFallback() local
DModelArgumentInfo.cpp72 const Operand& operand, const ANeuralNetworksOperandType* type, uint32_t poolIndex, in createFromMemory()
DExecutionBuilder.h119 std::optional<RunTimePoolInfo> getRunTimePoolInfo(uint32_t poolIndex) const { in getRunTimePoolInfo()
DModelBuilder.cpp297 uint32_t poolIndex = mMemories.add(mLargeValueMemory.get()); in copyLargeValuesToSharedMemory() local
/frameworks/ml/nn/common/
DBufferTracker.cpp61 ErrorStatus ManagedBuffer::validateRequest(uint32_t poolIndex, const Request& request, in validateRequest()
DCpuExecutor.cpp715 auto poolIndex = from.location.poolIndex; in initializeRunTimeInfo() local
762 auto poolIndex = from.location.poolIndex; in updateForArguments() local
/frameworks/ml/nn/driver/sample/
DSampleDriver.cpp414 const uint32_t poolIndex = request.outputs[i].location.poolIndex; in updateDeviceMemories() local
423 const uint32_t poolIndex = request.outputs[i].location.poolIndex; in updateDeviceMemories() local
434 const uint32_t poolIndex = request.outputs[i].location.poolIndex; in updateDeviceMemories() local
724 const uint32_t poolIndex = output.location.poolIndex; in executeFenced() local