mirror of
https://github.com/tensorflow/haskell.git
synced 2024-11-23 03:19:44 +01:00
2c5c879037
This change adds a class that both `Build` and `Session` are instances of:
class MonadBuild m where
build :: Build a -> m a
All stateful ops (generated and manually written) now have a signature that returns
an instance of `MonadBuild` (rather than just `Build`). For example:
assign_ :: (MonadBuild m, TensorType t)
=> Tensor Ref t -> Tensor v t -> m (Tensor Ref t)
This lets us remove a bunch of spurious calls to `build` in user code. It also
lets us replace the pattern `buildAnd run foo` with the simpler pattern `foo >>= run`
(or `run =<< foo`, which is sometimes nicer when foo is a complicated expression).
I went ahead and deleted `buildAnd` altogether since it seems to lead to
confusion; in particular a few tests had `buildAnd run . pure` which is
actually equivalent to just `run`.
|
||
|---|---|---|
| .. | ||
| ArrayOpsTest.hs | ||
| BuildTest.hs | ||
| DataFlowOpsTest.hs | ||
| EmbeddingOpsTest.hs | ||
| FeedFetchBench.hs | ||
| GradientTest.hs | ||
| MiscTest.hs | ||
| OpsTest.hs | ||
| RegressionTest.hs | ||
| TracingTest.hs | ||
| TypesTest.hs | ||