Home
last modified time | relevance | path

Searched refs:mOutputs (Results 1 – 3 of 3) sorted by relevance

/packages/modules/NeuralNetworks/runtime/
DExecutionBuilder.cpp158 mOutputs(mModel->outputCount()) { in ExecutionBuilder()
160 << " inputs and " << mOutputs.size() << " outputs"; in ExecutionBuilder()
266 uint32_t count = static_cast<uint32_t>(mOutputs.size()); in setOutput()
280 if (!mOutputs[index].unspecified()) { in setOutput()
286 std::tie(n, mOutputs[index]) = ModelArgumentInfo::createFromPointer( in setOutput()
302 uint32_t count = static_cast<uint32_t>(mOutputs.size()); in setOutputFromMemory()
326 if (!mOutputs[index].unspecified()) { in setOutputFromMemory()
332 std::tie(n, mOutputs[index]) = in setOutputFromMemory()
491 uint32_t count = static_cast<uint32_t>(mOutputs.size()); in getOutputOperandDimensions()
497 const auto& dims = mOutputs[index].dimensions(); in getOutputOperandDimensions()
[all …]
DExecutionBuilder.h134 const ModelArgumentInfo& getOutputInfo(uint32_t index) const { return mOutputs[index]; } in getOutputInfo()
187 std::vector<ModelArgumentInfo> mOutputs; variable
356 mapInputOrOutput(mExecutionBuilder->mOutputs[builderIndex], &mOutputs[executorIndex]); in mapOutput()
360 mapInputOrOutput(mExecutionBuilder->mOutputs[builderIndex], &mInputs[executorIndex], in mapOutputToInput()
377 length, dimensions, &mOutputs.at(outputIndex));
448 std::vector<ModelArgumentInfo> mOutputs; variable
/packages/modules/NeuralNetworks/shim_and_sl/include/
DSupportLibraryWrapper.h173 mOutputs = std::move(other.mOutputs);
285 mOutputs = outputs; in identifyInputsAndOutputs()
328 const std::vector<uint32_t>& getOutputs() const { return mOutputs; } in getOutputs()
339 std::vector<uint32_t> mOutputs; variable