Android NN API CpuExecutor: недопустимый тип или размеры для вывода модели

Я пытаюсь реализовать простую двухуровневую сеть в Android NN API, но при компиляции выдает ошибку:

E/CpuExecutor: Invalid type or dimensions for model output SOFTMAX failed.

E/CpuExecutor: Invalid type or dimensions for model output SOFTMAX failed.

может кто знает с чем связана эта ошибка? P / S / Есть ли у кого-нибудь рабочий пример реализации в NN API в репо? был бы очень признателен