Home
last modified time | relevance | path

Searched refs:SpatialDimensionOutputSizeAndPadding (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dconv_grad_size_util.cc24 StatusOr<SpatialDimensionOutputSizeAndPadding> GetWindowedOutputSize( in GetWindowedOutputSize()
37 SpatialDimensionOutputSizeAndPadding dim; in GetWindowedOutputSize()
66 StatusOr<SpatialDimensionOutputSizeAndPadding>
70 TF_ASSIGN_OR_RETURN(SpatialDimensionOutputSizeAndPadding output_dim, in ConvGradExtractAndVerifyDimension()
81 SpatialDimensionOutputSizeAndPadding dim; in ConvGradExtractAndVerifyDimension()
Dconv_grad_size_util.h26 struct SpatialDimensionOutputSizeAndPadding { struct
37 StatusOr<SpatialDimensionOutputSizeAndPadding> argument
Dpooling.cc257 SpatialDimensionOutputSizeAndPadding conv_backprop_spatial_dim, in AvgPoolGrad()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc2436 ::xla::SpatialDimensionOutputSizeAndPadding &conv_grad_spatial_dim = in matchAndRewrite()