Lines Matching refs:ModelArgumentInfo

156     std::tie(n, mInputs[index]) = ModelArgumentInfo::createFromPointer(  in setInput()
201 std::tie(n, mInputs[index]) = ModelArgumentInfo::createFromMemory( in setInputFromMemory()
234 ModelArgumentInfo::createFromPointer(mModel->getOutputOperand(index), type, buffer, l); in setOutput()
277 std::tie(n, mOutputs[index]) = ModelArgumentInfo::createFromMemory( in setOutputFromMemory()
747 if (p.state() == ModelArgumentInfo::UNSPECIFIED) { in computeFenced()
754 if (p.state() == ModelArgumentInfo::UNSPECIFIED) { in computeFenced()
761 if (mOutputs[i].state() != ModelArgumentInfo::HAS_NO_VALUE && in computeFenced()
804 if (p.state() == ModelArgumentInfo::UNSPECIFIED) { in compute()
807 } else if (p.state() == ModelArgumentInfo::MEMORY) { in compute()
815 if (p.state() == ModelArgumentInfo::UNSPECIFIED) { in compute()
878 if (x.state() != ModelArgumentInfo::HAS_NO_VALUE) { in getInitialOutputShapes()
918 if (output.state() != ModelArgumentInfo::MEMORY) continue; in updateMemories()
935 if (output.state() != ModelArgumentInfo::MEMORY) continue; in finishWithoutSyncFence()
999 void StepExecutor::mapInputOrOutput(const ModelArgumentInfo& builderInputOrOutput, in mapInputOrOutput()
1000 ModelArgumentInfo* executorInputOrOutput) { in mapInputOrOutput()
1006 case ModelArgumentInfo::HAS_NO_VALUE: in mapInputOrOutput()
1007 case ModelArgumentInfo::POINTER: in mapInputOrOutput()
1008 case ModelArgumentInfo::UNSPECIFIED: in mapInputOrOutput()
1010 case ModelArgumentInfo::MEMORY: { in mapInputOrOutput()
1022 ModelArgumentInfo* inputOrOutputInfo) { in setInputOrOutputFromMemory()
1032 ModelArgumentInfo::createFromMemory(inputOrOutputOperand, in setInputOrOutputFromMemory()
1037 static void logArguments(const char* kind, const std::vector<ModelArgumentInfo>& args) { in logArguments()
1042 case ModelArgumentInfo::POINTER: in logArguments()
1045 case ModelArgumentInfo::MEMORY: in logArguments()
1050 case ModelArgumentInfo::HAS_NO_VALUE: in logArguments()
1053 case ModelArgumentInfo::UNSPECIFIED: in logArguments()
1152 if (input.state() == ModelArgumentInfo::MEMORY) { in computeOnCpuFallback()
1158 if (output.state() == ModelArgumentInfo::MEMORY) { in computeOnCpuFallback()