From 8df7f120492b0b4a0af3f4e1c6a0ffe66ed63e37 Mon Sep 17 00:00:00 2001 From: Vijay Vasudevan Date: Mon, 31 Oct 2016 17:14:04 -0800 Subject: Make the DimensionHandle version of computing windowed output size public. Change: 137774322 --- tensorflow/core/framework/common_shape_fns.cc | 3 --- tensorflow/core/framework/common_shape_fns.h | 7 +++++++ 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/tensorflow/core/framework/common_shape_fns.cc b/tensorflow/core/framework/common_shape_fns.cc index f3a71a73f7..cc6cd10a08 100644 --- a/tensorflow/core/framework/common_shape_fns.cc +++ b/tensorflow/core/framework/common_shape_fns.cc @@ -70,8 +70,6 @@ Status Get3dOutputSize(const std::array& input, namespace shape_inference { -namespace { - Status GetWindowedOutputSizeFromDims( shape_inference::InferenceContext* c, shape_inference::DimensionHandle input_size, @@ -97,7 +95,6 @@ Status GetWindowedOutputSizeFromDims( } return Status::OK(); } -} // namespace Status UnchangedShape(shape_inference::InferenceContext* c) { c->set_output(0, c->input(0)); diff --git a/tensorflow/core/framework/common_shape_fns.h b/tensorflow/core/framework/common_shape_fns.h index 2e711f5277..176ea3519d 100644 --- a/tensorflow/core/framework/common_shape_fns.h +++ b/tensorflow/core/framework/common_shape_fns.h @@ -96,6 +96,13 @@ Status Get3dOutputSize(const std::array& input, namespace shape_inference { +// Like GetWindowedOutputSize, but deals with DimensionHandles. +Status GetWindowedOutputSizeFromDims(InferenceContext* c, + DimensionHandle input_size, + DimensionOrConstant filter_size, + int64 stride, Padding padding_type, + DimensionHandle* output_size); + // Transfers shape of input(0) to output(0). Status UnchangedShape(shape_inference::InferenceContext* c); -- cgit v1.2.3