From d684ef2f408fe7cd6c1b55a5ecdae3d7c4198bb0 Mon Sep 17 00:00:00 2001 From: Sean Silva Date: Tue, 28 Jul 2020 19:40:25 -0700 Subject: [PATCH] Make comments consistent on GroupedFilterShapeForDepthwiseConvolution PiperOrigin-RevId: 323704723 Change-Id: I00d6fb12ea053a6a8826a976833d752718980e34 --- tensorflow/compiler/tf2xla/kernels/conv_op_helpers.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/compiler/tf2xla/kernels/conv_op_helpers.cc b/tensorflow/compiler/tf2xla/kernels/conv_op_helpers.cc index e0bc2ba5052..a7a8b8bcb52 100644 --- a/tensorflow/compiler/tf2xla/kernels/conv_op_helpers.cc +++ b/tensorflow/compiler/tf2xla/kernels/conv_op_helpers.cc @@ -44,7 +44,7 @@ namespace tensorflow { namespace { // Returns the expanded size of a filter used for depthwise convolution. -// If `shape` is [H, W, ..., M, N] returns [H, W, ..., M, M*N]. +// If `shape` is [H, W, ..., M, N] returns [H, W, ..., 1, M*N]. xla::Shape GroupedFilterShapeForDepthwiseConvolution( const xla::Shape& filter_shape) { int64 input_feature_dim = filter_shape.dimensions_size() - 2; @@ -52,7 +52,7 @@ xla::Shape GroupedFilterShapeForDepthwiseConvolution( int64 depthwise_multiplier = filter_shape.dimensions(output_feature_dim); int64 input_feature = filter_shape.dimensions(input_feature_dim); - // Create a [H, W, ..., 1, N*M] reshape of the filter. + // Create a [H, W, ..., 1, M*N] reshape of the filter. xla::Shape grouped_filter_shape = filter_shape; grouped_filter_shape.set_dimensions(input_feature_dim, 1); grouped_filter_shape.set_dimensions(output_feature_dim,