wip: add test coverage for umd loss layer presenter