Currently we cannot run with heterogeneous backends, set all backends as
ACL. Later, we are going to control it with environment variables.
Signed-off-by: Hanjoung Lee <hanjoung.lee@samsung.com>
BackendResolver(::internal::BackendManager &backend_manager)
{
auto acl_gen = backend_manager.get("arm_compute");
- auto cpu_gen = backend_manager.get("cpu");
+ // auto cpu_gen = backend_manager.get("cpu"); // Unused for now
// TODO Set generator map according to environment variable
- _gen_map[typeid(::internal::tflite::op::Conv2D::implicit::Node)] = cpu_gen;
+ _gen_map[typeid(::internal::tflite::op::Conv2D::implicit::Node)] = acl_gen;
_gen_map[typeid(::internal::tflite::op::MaxPool2D::implicit::Node)] = acl_gen;
_gen_map[typeid(::internal::tflite::op::AvgPool2D::implicit::Node)] = acl_gen;
_gen_map[typeid(::internal::tflite::op::Concat::Node)] = acl_gen;