Searched refs:ModelArgumentInfo (Results 1 – 2 of 2) sorted by relevance
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()[all …]
43 struct ModelArgumentInfo { struct101 std::vector<ModelArgumentInfo> mInputs;102 std::vector<ModelArgumentInfo> mOutputs;169 int allocatePointerArgumentsToPool(std::vector<ModelArgumentInfo>* args, Memory* memory);172 void mapInputOrOutput(const ModelArgumentInfo& builderInputOrOutput,173 ModelArgumentInfo* executorInputOrOutput);177 ModelArgumentInfo* inputOrOutputInfo);198 std::vector<ModelArgumentInfo> mInputs;199 std::vector<ModelArgumentInfo> mOutputs;