public:
BackendResolver(::internal::BackendManager &backend_manager)
{
-#define OP(InternalName, NnApiName) \
- { \
- const auto &backend_str = \
- ::nnfw::util::EnvVar{std::string("OP_BACKEND_") + #NnApiName}.asString("arm_compute"); \
- auto backend = backend_manager.get(backend_str); \
- _gen_map[typeid(::internal::tflite::op::InternalName::Node)] = backend; \
+#define OP(InternalName, NnApiName) \
+ { \
+ const auto &backend_str = \
+ ::nnfw::util::EnvVar{std::string("OP_BACKEND_") + #NnApiName}.asString("acl_cl"); \
+ auto backend = backend_manager.get(backend_str); \
+ _gen_map[typeid(::internal::tflite::op::InternalName::Node)] = backend; \
}
#include "internal/op/Op.lst"
auto acl_stage_gen =
std::make_shared<::internal::arm_compute::StageGenerator>(operands, acl_tensor_builder);
- _gen_map["arm_compute"] = {acl_initializer_gen, acl_stage_gen};
+ // TODO Do not use magic string for backend id
+ _gen_map["acl_cl"] = {acl_initializer_gen, acl_stage_gen};
}
// Add CPU backend
auto cpu_stage_gen =
std::make_shared<::internal::cpu::StageGenerator>(operands, cpu_tensor_builder);
+ // TODO Do not use magic string for backend id
_gen_map["cpu"] = {cpu_initializer_gen, cpu_stage_gen};
}
}