diff options
author | 2016-10-31 17:14:04 -0800 | |
---|---|---|
committer | 2016-10-31 18:20:33 -0700 | |
commit | 8df7f120492b0b4a0af3f4e1c6a0ffe66ed63e37 (patch) | |
tree | 9973b5679bec4d5455275180e8c7b938b1a3a9e4 | |
parent | a74e4e72e42240c23fba3a9aae7015cc0c89ab6e (diff) |
Make the DimensionHandle version of computing windowed output size
public.
Change: 137774322
-rw-r--r-- | tensorflow/core/framework/common_shape_fns.cc | 3 | ||||
-rw-r--r-- | 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<int64, 3>& 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<int64, 3>& 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); |