Split embedding rknn method into preparation and execution part