diff options
| author |  Vijay Vasudevan <vrv@google.com> | 2016-07-21 14:39:24 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-07-21 15:48:02 -0700 |
| commit | 2641ef0c440161c8f75683d29d4ec53c52de51ed (patch) | |
| tree | 1dad1e62da195a68bc42374f231a8b559ec20a64 /tensorflow/core/framework/common_shape_fns.h | |
| parent | 68eaff83b1621a8c3cc67e8c940f9cf8515ac5fa (diff) | |
Add ShapeInference functions for DepthwiseConv2DNative, and Multiply
support to ShapeInference in support of DepthwiseConv.
Change: 128114077
Diffstat (limited to 'tensorflow/core/framework/common_shape_fns.h')
| -rw-r--r-- | tensorflow/core/framework/common_shape_fns.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/core/framework/common_shape_fns.h b/tensorflow/core/framework/common_shape_fns.h index 20410d87be..1926277395 100644 --- a/tensorflow/core/framework/common_shape_fns.h +++ b/tensorflow/core/framework/common_shape_fns.h @@ -157,6 +157,9 @@ Status BiasAddGradShape(shape_inference::InferenceContext* c); // Shape function for Conv2D-like operations. Status Conv2DShape(shape_inference::InferenceContext* c); +// Shape function for DepthwiseConv2D-like operations. +Status DepthwiseConv2DNativeShape(shape_inference::InferenceContext* c); + } // namespace shape_inference } // namespace tensorflow |