Implemented memory swapping heuristics for GPU
authorBenoit Steiner <bsteiner@google.com>
Fri, 5 Jan 2018 21:35:03 +0000 (13:35 -0800)
committerTensorFlower Gardener <gardener@tensorflow.org>
Fri, 5 Jan 2018 21:38:54 +0000 (13:38 -0800)
commitca6f0dd19b127c08e5e2cd7d0e6e9240881e6afa
treed22a1a5fa2a40b27612ecd513c74417799c5ec23
parent3a3feb207d8e138b7a468ae5d6e0d2daf4c8a49c
Implemented memory swapping heuristics for GPU

PiperOrigin-RevId: 180968225
tensorflow/core/grappler/costs/graph_memory.cc
tensorflow/core/grappler/graph_view.h
tensorflow/core/grappler/optimizers/memory_optimizer.cc
tensorflow/core/grappler/optimizers/memory_optimizer_test.cc