mirror of
https://github.com/tensorflow/haskell.git
synced 2024-06-03 03:23:37 +02: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. |
||
|---|---|---|
| .. | ||
| ArrayOpsTest.hs | ||
| BuildTest.hs | ||
| DataFlowOpsTest.hs | ||
| EmbeddingOpsTest.hs | ||
| FeedFetchBench.hs | ||
| GradientTest.hs | ||
| MiscTest.hs | ||
| OpsTest.hs | ||
| RegressionTest.hs | ||
| TracingTest.hs | ||
| TypesTest.hs | ||