@@ -144,7 +144,7 @@ tsl::Status GetWindowedOutputSizeVerboseV2(
144
144
145
145
// Check dimension
146
146
tsl::Status ConvBackpropExtractAndVerifyDimension (
147
- tsl::StringPiece label, const xla::Shape& input_shape,
147
+ absl::string_view label, const xla::Shape& input_shape,
148
148
const xla::Shape& filter_shape, const xla::Shape& output_shape,
149
149
const absl::Span<const tsl::int32> dilations,
150
150
const std::vector<tsl::int32>& strides, Padding padding,
@@ -185,7 +185,7 @@ tsl::Status ConvBackpropExtractAndVerifyDimension(
185
185
186
186
// Check dimension
187
187
tsl::Status ConvBackpropComputeDimensionsV2 (
188
- tsl::StringPiece label, int num_spatial_dims, const xla::Shape& input_shape,
188
+ absl::string_view label, int num_spatial_dims, const xla::Shape& input_shape,
189
189
const xla::Shape& filter_shape, const xla::Shape& out_backprop_shape,
190
190
absl::Span<const tsl::int32> dilations,
191
191
const std::vector<tsl::int32>& strides, Padding padding,
@@ -293,7 +293,7 @@ xla::XlaOp TransposeFilterForGroupConvolutionBackpropInput(
293
293
294
294
// Wrapper for ConvGeneralDilated and check dim.
295
295
tsl::StatusOr<xla::XlaOp> MakeXlaBackpropInputConvOp (
296
- tsl::StringPiece type_string, const xla::Shape& input_shape,
296
+ absl::string_view type_string, const xla::Shape& input_shape,
297
297
xla::XlaOp filter, xla::XlaOp out_backprop, const ConvOpAttrs& attrs,
298
298
xla::XlaOp* input_sizes) {
299
299
TF_RETURN_IF_ERROR (CheckConvAttrs (attrs));
@@ -394,7 +394,7 @@ tsl::StatusOr<xla::XlaOp> MakeXlaBackpropInputConvOp(
394
394
395
395
// Wrapper for ConvGeneralDilated and check dim.
396
396
tsl::StatusOr<xla::XlaOp> MakeXlaBackpropFilterConvOp (
397
- tsl::StringPiece type_string, xla::XlaOp activations,
397
+ absl::string_view type_string, xla::XlaOp activations,
398
398
const xla::Shape& filter_shape, xla::XlaOp gradients,
399
399
const ConvOpAttrs& attrs) {
400
400
TF_RETURN_IF_ERROR (CheckConvAttrs (attrs));
@@ -551,4 +551,4 @@ tsl::StatusOr<xla::XlaOp> MakeXlaBackpropFilterConvOp(
551
551
552
552
// -------------Convolution Helper Function End--------------------------
553
553
554
- } // namespace torch_xla
554
+ } // namespace torch_xla
0 commit comments