Searched refs:ioType (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/runtime/ |
| D | Memory.cpp | 104 bool validate(const CompilationBuilder* compilation, IOType ioType, uint32_t index, in validate() argument
107 NN_RET_CHECK(kCompilationRoles.count({compilation, ioType, index}) > 0) in validate()
324 int MemoryBuilder::addRole(const CompilationBuilder& compilation, IOType ioType, uint32_t index, in addRole() argument
326 const char* tag = ioType == IOType::INPUT ? "addInputRole" : "addOutputRole"; in addRole()
330 if (mRoles.count({&compilation, ioType, index}) > 0) { in addRole()
340 if (ioType == IOType::INPUT) { in addRole()
353 if (ioType == IOType::INPUT) { in addRole()
390 mRoles.emplace(&compilation, ioType, index); in addRole()
|
| D | Memory.h | 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);
|