提交 8df7f120 编写于 作者: V Vijay Vasudevan 提交者: TensorFlower Gardener

Make the DimensionHandle version of computing windowed output size

public.
Change: 137774322
上级 a74e4e72
......@@ -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));
......
......@@ -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);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册