tensorflow/compiler/tf2xla/kernels/conv_op_helpers.cc
// Wrapper around ConvBackpropComputeDimensions that converts from XLA shapes
// to TensorShapes.
Wrapper around ConvBackpropComputeDimensions that converts from XLA shapes
to TensorShapes.
Wrapper around ConvBackpropComputeDimensions that converts from XLA shapes to TensorShapes .
Status ConvBackpropComputeDimensionsV2XlaShapes(
    StringPiece label, int num_spatial_dims, const xla::Shape& input_shape,
    const xla::Shape& filter_shape, const xla::Shape& out_backprop_shape,
    absl::Span<const int32> dilations, const std::vector<int32>& strides,
    Padding padding, TensorFormat data_format, ConvBackpropDimensions* dims,
    absl::Span<const int64_t> explicit_paddings) {
  TensorShape input_tensor_shape, filter_tensor_shape,
      out_backprop_tensor_shape;
  TF_RETURN_IF_ERROR(XLAShapeToTensorShape(input_shape, &input_tensor_shape));
  TF_RETURN_IF_ERROR(XLAShapeToTensorShape(filter_shape, &filter_tensor_shape));
  TF_RETURN_IF_ERROR(
      XLAShapeToTensorShape(out_backprop_shape, &out_backprop_tensor_shape));
  return ConvBackpropComputeDimensionsV2(
      label, num_spatial_dims, input_tensor_shape, filter_tensor_shape,
      out_backprop_tensor_shape, dilations, strides, padding, explicit_paddings,
      data_format, dims);
}
Status ConvBackpropComputeDimensionsV2XlaShapes ( StringPiece label , int num_spatial_dims , const xla :: Shape & input_shape , const xla :: Shape & filter_shape , const xla :: Shape & out_backprop_shape , absl :: Span < const int32 > dilations , const std :: vector < int32 > & strides , Padding padding , TensorFormat data_format , ConvBackpropDimensions * dims , absl :: Span < const int64_t > explicit_paddings ) { TensorShape input_tensor_shape , filter_tensor_shape , out_backprop_tensor_shape ; TF_RETURN_IF_ERROR ( XLAShapeToTensorShape ( input_shape , & input_tensor_shape ) ) ; TF_RETURN_IF_ERROR ( XLAShapeToTensorShape ( filter_shape , & filter_tensor_shape ) ) ; TF_RETURN_IF_ERROR ( XLAShapeToTensorShape ( out_backprop_shape , & out_backprop_tensor_shape ) ) ; return ConvBackpropComputeDimensionsV2 ( label , num_spatial_dims , input_tensor_shape , filter_tensor_shape , out_backprop_tensor_shape , dilations , strides , padding , explicit_paddings , data_format , dims ) ; }
Wrapper around ConvBackpropComputeDimensions that converts from XLA shapes
to TensorShapes.
Wrapper around ConvBackpropComputeDimensions that converts from XLA shapes to TensorShapes .