diff options
Diffstat (limited to 'tensorflow/compiler/tf2xla/kernels/depthwise_conv_ops.cc')
| -rw-r--r-- | tensorflow/compiler/tf2xla/kernels/depthwise_conv_ops.cc | 19 |
1 files changed, 10 insertions, 9 deletions
diff --git a/tensorflow/compiler/tf2xla/kernels/depthwise_conv_ops.cc b/tensorflow/compiler/tf2xla/kernels/depthwise_conv_ops.cc index 852d2a966e..92b371cc4e 100644 --- a/tensorflow/compiler/tf2xla/kernels/depthwise_conv_ops.cc +++ b/tensorflow/compiler/tf2xla/kernels/depthwise_conv_ops.cc @@ -172,14 +172,15 @@ class DepthwiseConv2dNativeOp : public XlaOpKernel { } else { // These will be used to define the bounds of each slice. // Within the loop, the input_channel index will be modified. - gtl::InlinedVector<int64, 4> filter_begin(4, 0); - gtl::InlinedVector<int64, 4> filter_limits(4); - gtl::InlinedVector<int64, 4> input_begin(4, 0); - gtl::InlinedVector<int64, 4> input_limits(4); - gtl::InlinedVector<int64, 4> strides(4, 1); + gtl::InlinedVector<int64, 4> filter_begin; + gtl::InlinedVector<int64, 4> filter_limits; + gtl::InlinedVector<int64, 4> input_begin; + gtl::InlinedVector<int64, 4> input_limits; for (int i = 0; i < 4; ++i) { - filter_limits[i] = filter_shape.dim_size(i); - input_limits[i] = input_shape.dim_size(i); + filter_begin.push_back(0); + filter_limits.push_back(filter_shape.dim_size(i)); + input_begin.push_back(0); + input_limits.push_back(input_shape.dim_size(i)); } std::vector<int64> strides_for_tla{strides_[1], strides_[2]}; @@ -208,9 +209,9 @@ class DepthwiseConv2dNativeOp : public XlaOpKernel { input_limits[3] = i + 1; xla::ComputationDataHandle filter_slice = - b.Slice(filter, filter_begin, filter_limits, strides); + b.Slice(filter, filter_begin, filter_limits); xla::ComputationDataHandle input_slice = - b.Slice(input, input_begin, input_limits, strides); + b.Slice(input, input_begin, input_limits); convs.push_back(b.ConvWithGeneralDimensions( input_slice, filter_slice, strides_for_tla, xla_padding, dims)); } |