pkgs/development/haskell-modules/configuration-tensorflow.nix

   1 { pkgs, haskellLib }:
   2
   3 with haskellLib;
   4
   5 self: super:
   6 let
   7   # This contains updates to the dependencies, without which it would
   8   # be even more work to get it to build.
   9   # As of 2020-04, there's no new release in sight, which is why we're
  10   # pulling from Github.
  11   tensorflow-haskell = pkgs.fetchFromGitHub {
  12     owner = "tensorflow";
  13     repo = "haskell";
  14     rev = "568c9b6f03e5d66a25685a776386e2ff50b61aa9";
  15     sha256 = "0v58zhqipa441hzdvp9pwgv6srir2fm7cp0bq2pb5jl1imwyd37h";
  16     fetchSubmodules = true;
  17   };
  18
  19   setTensorflowSourceRoot = dir: drv:
  20     (overrideCabal drv (drv: { src = tensorflow-haskell; }))
  21       .overrideAttrs (_oldAttrs: {sourceRoot = "source/${dir}";});
  22 in
  23 {
  24   tensorflow-proto = doJailbreak (setTensorflowSourceRoot "tensorflow-proto" super.tensorflow-proto);
  25
  26   tensorflow = (setTensorflowSourceRoot "tensorflow" super.tensorflow).override {
  27     # the "regular" Python package does not seem to include the binary library
  28     libtensorflow = pkgs.libtensorflow-bin;
  29   };
  30
  31   tensorflow-core-ops = setTensorflowSourceRoot "tensorflow-core-ops" super.tensorflow-core-ops;
  32
  33   tensorflow-logging = setTensorflowSourceRoot "tensorflow-logging" super.tensorflow-logging;
  34
  35   tensorflow-opgen = setTensorflowSourceRoot "tensorflow-opgen" super.tensorflow-opgen;
  36
  37   tensorflow-ops = setTensorflowSourceRoot "tensorflow-ops" super.tensorflow-ops;
  38 }