mirror of
https://github.com/tensorflow/haskell.git
synced 2024-12-27 20:19:46 +01:00
c99a23b6a7
Each op `foo :: ...` now has a corresponding `foo' :: OpParams -> ...` which lets you set optional attributes. `OpParams` is currently a type alias for `OpDef -> OpDef`. In the future we should consider more type safety, e.g., using type-level strings and OverloadedLabels for optional attributes. I used it to replace a few manual `buildOp`s in our code with the codegenerated ops, now that it's easier to set attributes. I also removed `tensorAttr` and `named` since it's now possible to set those op attributes directly. Although this clutters up the API a bit, I think it's simpler than using type classes to implement optional arguments (as in, for example, `Text.Printf`) -- especially in terms of type inference with the rest of the library. |
||
---|---|---|
.. | ||
src/TensorFlow | ||
tests | ||
Setup.hs | ||
tensorflow-ops.cabal |