Lines Matching refs:ModelArgumentInfo
204 std::tie(n, mInputs[index]) = ModelArgumentInfo::createFromPointer( in setInput()
253 ModelArgumentInfo::createFromMemory(mModel->getInputOperand(index), type, poolIndex, in setInputFromMemory()
286 std::tie(n, mOutputs[index]) = ModelArgumentInfo::createFromPointer( in setOutput()
333 ModelArgumentInfo::createFromMemory(mModel->getOutputOperand(index), type, poolIndex, in setOutputFromMemory()
548 static int validateRequest(const std::vector<ModelArgumentInfo>& inputs, in validateRequest()
549 const std::vector<ModelArgumentInfo>& outputs) { in validateRequest()
551 if (p.state() == ModelArgumentInfo::UNSPECIFIED) { in validateRequest()
557 if (p.state() == ModelArgumentInfo::UNSPECIFIED) { in validateRequest()
576 if (mOutputs[i].state() != ModelArgumentInfo::HAS_NO_VALUE && in areOutputsFullySpecified()
1026 if (p.state() == ModelArgumentInfo::MEMORY) { in compute()
1095 if (x.state() != ModelArgumentInfo::HAS_NO_VALUE) { in getInitialOutputShapes()
1145 if (output.state() != ModelArgumentInfo::MEMORY) continue; in updateMemories()
1159 if (output.state() != ModelArgumentInfo::MEMORY) continue; in finishComputation()
1343 void StepExecutor::mapInputOrOutput(const ModelArgumentInfo& builderInputOrOutput, in mapInputOrOutput()
1344 ModelArgumentInfo* executorInputOrOutput, in mapInputOrOutput()
1358 case ModelArgumentInfo::HAS_NO_VALUE: in mapInputOrOutput()
1359 case ModelArgumentInfo::UNSPECIFIED: in mapInputOrOutput()
1361 case ModelArgumentInfo::POINTER: in mapInputOrOutput()
1364 case ModelArgumentInfo::MEMORY: { in mapInputOrOutput()
1378 ModelArgumentInfo* inputOrOutputInfo) { in setInputOrOutputFromMemory()
1387 ModelArgumentInfo::createFromMemory(inputOrOutputOperand, in setInputOrOutputFromMemory()
1411 static void logArguments(const char* kind, const std::vector<ModelArgumentInfo>& args) { in logArguments()
1416 case ModelArgumentInfo::POINTER: in logArguments()
1420 case ModelArgumentInfo::MEMORY: in logArguments()
1426 case ModelArgumentInfo::HAS_NO_VALUE: in logArguments()
1429 case ModelArgumentInfo::UNSPECIFIED: in logArguments()
1553 if (input.state() == ModelArgumentInfo::MEMORY) { in computeOnCpuFallback()
1559 if (output.state() == ModelArgumentInfo::MEMORY) { in computeOnCpuFallback()