Searched refs:window_strides_attr (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | shape_inference.cc | 1736 DenseIntElementsAttr window_strides_attr, padding_attr, lhs_dilation_attr, in InferShapeForXlaConvV2Op() local
1738 if (matchPattern(window_strides, m_Constant(&window_strides_attr)) && in InferShapeForXlaConvV2Op()
1763 for (const llvm::APInt& i : window_strides_attr) { in InferShapeForXlaConvV2Op()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 3399 DenseElementsAttr window_strides_attr, padding_attr, lhs_dilation_attr, in verify() local
3401 if (!(matchPattern(op.window_strides(), m_Constant(&window_strides_attr)) && in verify()
3409 if (window_strides_attr.getType().getRank() != 1) in verify()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1200 auto window_strides_attr = rewriter.getNamedAttr( in matchAndRewriteDynamicConv() local
1244 NamedAttribute attrs[] = {rhs_dilations_attr, window_strides_attr, in matchAndRewriteDynamicConv()
1355 auto window_strides_attr = rewriter.getNamedAttr( in matchAndRewrite() local
1399 NamedAttribute attrs[] = {rhs_dilations_attr, window_strides_attr, in matchAndRewrite()
6999 DenseElementsAttr window_strides_attr, padding_attr, lhs_dilation_attr, in matchAndRewrite() local
7001 if (!(matchPattern(op.window_strides(), m_Constant(&window_strides_attr)) && in matchAndRewrite()
7010 "window_strides", hlo::convertElementsAttr(window_strides_attr, in matchAndRewrite()
|