Searched refs:zero_attr (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 1142 auto zero_attr = rewriter.getZeroAttr(shaped_type.getElementType()); in matchAndRewrite() local
1143 Value zero = rewriter.create<ConstantOp>(loc, zero_attr); in matchAndRewrite()
1232 auto zero_attr = rewriter.getZeroAttr(shaped_type.getElementType()); in matchAndRewrite() local
1233 Value zero = rewriter.create<ConstantOp>(loc, zero_attr); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 2368 Attribute zero_attr; in matchAndRewrite() local
2370 zero_attr = rewriter.getZeroAttr(type); in matchAndRewrite()
2372 zero_attr = DenseStringElementsAttr::get(type, {""}); in matchAndRewrite()
2374 if (!zero_attr) return failure(); in matchAndRewrite()
2376 auto zero_const = rewriter.create<TF::ConstOp>(op.getLoc(), zero_attr); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | lower_static_tensor_list.cc | 473 Attribute zero_attr = rewriter.getZeroAttr(zero_type); in matchAndRewrite() local
474 auto zero = rewriter.create<ConstantOp>(loc, zero_type, zero_attr); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 430 Attribute zero_attr = rewriter.getZeroAttr(zero_type); in convertZerosLikeOp() local
434 zero_attr.cast<ElementsAttr>()) in convertZerosLikeOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5193 auto zero_attr = IntegerAttr::get(rewriter.getIntegerType(32), 0); in matchAndRewrite() local
5195 op.getLoc(), unpacked_indices_type, indices, zero_attr); in matchAndRewrite()
5208 rewriter.create<mhlo::ConstOp>(op.getLoc(), zero_attr).getResult(); in matchAndRewrite()
|