Searched refs:paddings_attr (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 1240 ElementsAttr paddings_attr = nullptr; in Verify() local
1257 if (matchPattern(op.paddings(), m_Constant(&paddings_attr))) { in Verify()
1260 paddings_attr.getValue({i, 0}).cast<IntegerAttr>().getInt(); in Verify()
1262 paddings_attr.getValue({i, 1}).cast<IntegerAttr>().getInt(); in Verify()
1273 if (input_type.hasStaticShape() && block_shape_attr && paddings_attr) { in Verify()
1277 paddings_attr.getValue({i, 0}).cast<IntegerAttr>().getInt(); in Verify()
1279 paddings_attr.getValue({i, 1}).cast<IntegerAttr>().getInt(); in Verify()
1337 DenseIntElementsAttr paddings_attr; in inferReturnTypes() local
1339 GetValueAsConstant(paddings_val, paddings_attr)) { in inferReturnTypes()
1351 paddings_attr.getValue<APInt>({i, 0}).getSExtValue() + in inferReturnTypes()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1258 auto paddings_attr = rewriter.getNamedAttr( in matchAndRewrite() local
1278 batch_group_count_attr, paddings_attr}; in matchAndRewrite()
2287 DenseIntElementsAttr paddings_attr = GetReduceWindowPaddingAsAttr<num_dims>( in matchAndRewrite() local
2294 /*window_dilations=*/DenseIntElementsAttr(), paddings_attr); in matchAndRewrite()
2530 DenseIntElementsAttr paddings_attr = GetReduceWindowPaddingAsAttr<num_dims>( in matchAndRewrite() local
2536 /*window_dilations=*/DenseIntElementsAttr(), paddings_attr); in matchAndRewrite()
4059 DenseIntElementsAttr paddings_attr = GetReduceWindowPaddingAsAttr<num_dims>( in matchAndRewrite() local
4066 paddings_attr); in matchAndRewrite()
4183 auto paddings_attr = DenseIntElementsAttr::get(paddings_ty, paddings); in matchAndRewrite() local
4242 /*padding=*/paddings_attr, GetI64ElementsAttr(lhs_dilation, &rewriter), in matchAndRewrite()
[all …]
|