forked from mirrors/nixpkgs
82a286599e
Its tensorflow-mnist-data dependency is not on Hackage, and the license conditions for the data are unclear. Also, it's basically an example package.
39 lines
1.3 KiB
Nix
39 lines
1.3 KiB
Nix
{ pkgs, haskellLib }:
|
|
|
|
with haskellLib;
|
|
|
|
self: super:
|
|
let
|
|
# This contains updates to the dependencies, without which it would
|
|
# be even more work to get it to build.
|
|
# As of 2020-04, there's no new release in sight, which is why we're
|
|
# pulling from Github.
|
|
tensorflow-haskell = pkgs.fetchFromGitHub {
|
|
owner = "tensorflow";
|
|
repo = "haskell";
|
|
rev = "568c9b6f03e5d66a25685a776386e2ff50b61aa9";
|
|
sha256 = "0v58zhqipa441hzdvp9pwgv6srir2fm7cp0bq2pb5jl1imwyd37h";
|
|
fetchSubmodules = true;
|
|
};
|
|
|
|
setTensorflowSourceRoot = dir: drv:
|
|
(overrideCabal drv (drv: { src = tensorflow-haskell; }))
|
|
.overrideAttrs (_oldAttrs: {sourceRoot = "source/${dir}";});
|
|
in
|
|
{
|
|
tensorflow-proto = doJailbreak (setTensorflowSourceRoot "tensorflow-proto" super.tensorflow-proto);
|
|
|
|
tensorflow = (setTensorflowSourceRoot "tensorflow" super.tensorflow).override {
|
|
# the "regular" Python package does not seem to include the binary library
|
|
libtensorflow = pkgs.libtensorflow-bin;
|
|
};
|
|
|
|
tensorflow-core-ops = setTensorflowSourceRoot "tensorflow-core-ops" super.tensorflow-core-ops;
|
|
|
|
tensorflow-logging = setTensorflowSourceRoot "tensorflow-logging" super.tensorflow-logging;
|
|
|
|
tensorflow-opgen = setTensorflowSourceRoot "tensorflow-opgen" super.tensorflow-opgen;
|
|
|
|
tensorflow-ops = setTensorflowSourceRoot "tensorflow-ops" super.tensorflow-ops;
|
|
}
|