diff options
author | 2016-10-14 17:04:20 -0800 | |
---|---|---|
committer | 2016-10-14 18:16:57 -0700 | |
commit | cd92a576f927329a2686c5456402ebd88e9ba710 (patch) | |
tree | 01b5d38c300aa465f8bf8b65bc419a9aa58823f7 /tensorflow/core/common_runtime/direct_session_with_tracking_alloc_test.cc | |
parent | 5d1c0c41510f51a707373f0d1efabf3be44fa480 (diff) |
Added the data type to the cost graph
Change: 136219453
Diffstat (limited to 'tensorflow/core/common_runtime/direct_session_with_tracking_alloc_test.cc')
-rw-r--r-- | tensorflow/core/common_runtime/direct_session_with_tracking_alloc_test.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/common_runtime/direct_session_with_tracking_alloc_test.cc b/tensorflow/core/common_runtime/direct_session_with_tracking_alloc_test.cc index 228345d52d..fbac512785 100644 --- a/tensorflow/core/common_runtime/direct_session_with_tracking_alloc_test.cc +++ b/tensorflow/core/common_runtime/direct_session_with_tracking_alloc_test.cc @@ -254,6 +254,8 @@ TEST(DirectSessionWithTrackingAllocTest, CostGraph) { EXPECT_EQ(2, shape.dim_size()); EXPECT_EQ(2, shape.dim(0).size()); EXPECT_EQ(1, shape.dim(1).size()); + const DataType& dtype = node.output_info(0).dtype(); + EXPECT_EQ(DT_FLOAT, dtype); } EXPECT_LE(0, node.compute_cost()); EXPECT_GE(run_duration_micros, node.compute_cost()); |