diff options
author | 2017-04-12 08:22:17 -0800 | |
---|---|---|
committer | 2017-04-12 09:45:16 -0700 | |
commit | a18d61d2f6507b29dc076a8711ff310c48d053ca (patch) | |
tree | a323e05c94653ebe474ed76194690026e3143bac | |
parent | e9d407a67bf6a68393f6f991dd4b076a648804c0 (diff) |
Switch the benchmark in direct_session_test.cc to use Placeholder (not V2).
Change: 152946496
-rw-r--r-- | tensorflow/core/common_runtime/direct_session_test.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/common_runtime/direct_session_test.cc b/tensorflow/core/common_runtime/direct_session_test.cc index c8b8a09b8e..f8deaaf222 100644 --- a/tensorflow/core/common_runtime/direct_session_test.cc +++ b/tensorflow/core/common_runtime/direct_session_test.cc @@ -1178,7 +1178,7 @@ void FeedFetchBenchmarkHelper(int num_feeds, int iters) { // monitor this overhead where possible, but that is not the // object of study in this benchmark. Node* placeholder; - TF_CHECK_OK(NodeBuilder(g.NewName("Placeholder"), "PlaceholderV2") + TF_CHECK_OK(NodeBuilder(g.NewName("Placeholder"), "Placeholder") .Attr("shape", TensorShape()) .Attr("dtype", DT_FLOAT) .Device("/cpu:0") |