* Add interpret_as_scalar helper

* Add interpret_as_scalar to OneHot

* clang-format

* Remove interpret_as_scalar from OneHot

We currently only support OneHot with constant depth

* Review comments