Home
last modified time | relevance | path

Searched refs:IOType (Results 1 – 9 of 9) sorted by relevance

/frameworks/ml/nn/runtime/
DMemory.cpp54 bool validate(const CompilationBuilder*, IOType, uint32_t, const ANeuralNetworksOperandType*, in validate() argument
77 bool validate(const CompilationBuilder* compilation, IOType, uint32_t, in validate() argument
104 bool validate(const CompilationBuilder* compilation, IOType ioType, uint32_t index, in validate()
324 int MemoryBuilder::addRole(const CompilationBuilder& compilation, IOType ioType, uint32_t index, in addRole()
326 const char* tag = ioType == IOType::INPUT ? "addInputRole" : "addOutputRole"; in addRole()
336 std::vector<std::tuple<const PreparedModel*, IOType, uint32_t>> roles; in addRole()
337 auto callback = [&roles](const auto* preparedModel, IOType type, uint32_t index) { in addRole()
340 if (ioType == IOType::INPUT) { in addRole()
353 if (ioType == IOType::INPUT) { in addRole()
394 if (type == IOType::INPUT) { in addRole()
DMemory.h95 using CompilationRole = std::tuple<const CompilationBuilder*, IOType, uint32_t>;
96 using StepRoleCallback = std::function<void(const PreparedModel*, IOType, uint32_t)>;
128 virtual bool validate(const CompilationBuilder* compilation, IOType ioType, uint32_t index,
236 int addRole(const CompilationBuilder& compilation, IOType ioType, uint32_t index, float freq);
DExecutionPlan.cpp1547 callback(mPreparedModel.get(), IOType::INPUT, index); in forEachStepRoleOfInput()
1552 callback(mPreparedModel.get(), IOType::OUTPUT, index); in forEachStepRoleOfOutput()
1566 callback(step->getPreparedStepModel().get(), IOType::INPUT, i); in forEachStepRoleOfInput()
1586 callback(step->getPreparedStepModel().get(), IOType::OUTPUT, i); in forEachStepRoleOfOutput()
1596 callback(step->getPreparedStepModel().get(), IOType::INPUT, i); in forEachStepRoleOfOutput()
DExecutionBuilder.cpp180 if (!memory->getValidator().validate(mCompilation, IOType::INPUT, index, type, offset, in setInputFromMemory()
257 if (!memory->getValidator().validate(mCompilation, IOType::OUTPUT, index, type, offset, in setOutputFromMemory()
DNeuralNetworks.cpp928 return mb->addRole(*c, IOType::INPUT, index, frequency); in ANeuralNetworksMemoryDesc_addInputRole()
941 return mb->addRole(*c, IOType::OUTPUT, index, frequency); in ANeuralNetworksMemoryDesc_addOutputRole()
DModelBuilder.cpp338 …if (!memory->getValidator().validate(/*compilation=*/nullptr, /*placeholder*/ IOType::INPUT, index, in setOperandValueFromMemory()
/frameworks/ml/nn/common/include/
DValidateHal.h37 enum class IOType { INPUT, OUTPUT }; enum
38 using PreparedModelRole = std::tuple<const hal::IPreparedModel*, IOType, uint32_t>;
/frameworks/ml/nn/common/
DBufferTracker.cpp73 if (kRoles.count({preparedModel, IOType::INPUT, i}) == 0) { in validateRequest()
100 if (kRoles.count({preparedModel, IOType::OUTPUT, i}) == 0) { in validateRequest()
DValidateHal.cpp877 const auto [it, success] = roles.emplace(preparedModel.get(), IOType::INPUT, role.ioIndex); in validateMemoryDesc()
891 const auto [it, success] = roles.emplace(preparedModel.get(), IOType::OUTPUT, role.ioIndex); in validateMemoryDesc()