Lines Matching refs:ModelArgumentInfo
35 int ModelArgumentInfo::setFromPointer(const Operand& operand, in setFromPointer()
45 state = ModelArgumentInfo::HAS_NO_VALUE; in setFromPointer()
57 state = ModelArgumentInfo::POINTER; in setFromPointer()
64 int ModelArgumentInfo::setFromMemory(const Operand& operand, const ANeuralNetworksOperandType* type, in setFromMemory()
77 state = ModelArgumentInfo::MEMORY; in setFromMemory()
83 int ModelArgumentInfo::setFromTemporaryMemory(const Operand& operand, in setFromTemporaryMemory()
89 state = ModelArgumentInfo::MEMORY; in setFromTemporaryMemory()
96 int ModelArgumentInfo::updateDimensionInfo(const Operand& operand, in updateDimensionInfo()
334 if (p.state == ModelArgumentInfo::UNSPECIFIED) { in startCompute()
340 if (p.state == ModelArgumentInfo::UNSPECIFIED) { in startCompute()
418 int StepExecutor::allocatePointerArgumentsToPool(std::vector<ModelArgumentInfo>* args, in allocatePointerArgumentsToPool()
423 if (info.state == ModelArgumentInfo::POINTER) { in allocatePointerArgumentsToPool()
445 static void setRequestArgumentArray(const std::vector<ModelArgumentInfo>& argumentInfos, in setRequestArgumentArray()
451 (*ioInfos)[i] = { .hasNoValue = info.state == ModelArgumentInfo::HAS_NO_VALUE, in setRequestArgumentArray()
471 void StepExecutor::mapInputOrOutput(const ModelArgumentInfo& builderInputOrOutput, in mapInputOrOutput()
472 ModelArgumentInfo* executorInputOrOutput) { in mapInputOrOutput()
477 case ModelArgumentInfo::POINTER: in mapInputOrOutput()
478 case ModelArgumentInfo::UNSPECIFIED: in mapInputOrOutput()
480 case ModelArgumentInfo::MEMORY: { in mapInputOrOutput()
494 ModelArgumentInfo* inputOrOutputInfo) { in setInputOrOutputFromTemporaryMemory()
503 static void logArguments(const char* kind, const std::vector<ModelArgumentInfo> &args) { in logArguments()
508 case ModelArgumentInfo::POINTER: in logArguments()
511 case ModelArgumentInfo::MEMORY: in logArguments()
518 case ModelArgumentInfo::HAS_NO_VALUE: in logArguments()
521 case ModelArgumentInfo::UNSPECIFIED: in logArguments()
600 if (info.state == ModelArgumentInfo::POINTER) { in startComputeOnDevice()
665 if (info.state == ModelArgumentInfo::POINTER) { in startComputeOnDevice()
718 auto fixPointerArguments = [&requestPoolInfos](std::vector<ModelArgumentInfo>& argumentInfos) { in startComputeOnCpu()
719 for (ModelArgumentInfo& argumentInfo : argumentInfos) { in startComputeOnCpu()
720 if (argumentInfo.state == ModelArgumentInfo::POINTER) { in startComputeOnCpu()