| 1192 | } |
| 1193 | |
| 1194 | XlaOp XlaBuilder::Conv(XlaOp lhs, XlaOp rhs, |
| 1195 | absl::Span<const int64> window_strides, Padding padding, |
| 1196 | int64 feature_group_count, int64 batch_group_count, |
| 1197 | const PrecisionConfig* precision_config) { |
| 1198 | return ConvWithGeneralDimensions( |
| 1199 | lhs, rhs, window_strides, padding, |
| 1200 | CreateDefaultConvDimensionNumbers(window_strides.size()), |
| 1201 | feature_group_count, batch_group_count, precision_config); |
| 1202 | } |
| 1203 | |
| 1204 | XlaOp XlaBuilder::ConvWithGeneralPadding( |
| 1205 | XlaOp lhs, XlaOp rhs, absl::Span<const int64> window_strides, |
no test coverage detected