Skip to content

[utils]

Utils for tensorflow backend support code.

GPU/CPU backing device management utils

Loss function management utils

  • build_keras_loss: Type-check, deserialize and/or wrap a keras loss into a Loss object.

Better support for sparse tensor structures