Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc2466 DenseIntElementsAttr multiples_attr; in verify() local
2467 if (matchPattern(op.multiples(), m_Constant(&multiples_attr))) { in verify()
2471 const int64_t m = multiples_attr.getValues<APInt>()[i].getSExtValue(); in verify()
2494 DenseIntElementsAttr multiples_attr; in fold() local
2495 if (matchPattern(multiples(), m_Constant(&multiples_attr))) { in fold()
2498 if (multiples_attr.isSplat() && in fold()
2499 multiples_attr.getSplatValue<APInt>().getSExtValue() == 1) { in fold()
/external/tensorflow/tensorflow/core/transforms/constant_folding/
Dpass.cc1700 ElementsAttr multiples_attr = in matchAndRewrite() local
1702 if (multiples_attr.getElementType().isInteger(32)) { in matchAndRewrite()
1703 if (llvm::any_of(multiples_attr.getValues<int32_t>(), in matchAndRewrite()
1708 if (llvm::any_of(multiples_attr.getValues<int64_t>(), in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc1467 ArrayAttr multiples_attr = rewriter.getI64ArrayAttr(multiples_vals); in matchAndRewrite() local
1470 tf_tile_op.input(), multiples_attr); in matchAndRewrite()
Dlegalize_tfl.cc2084 ArrayAttr multiples_attr = rewriter.getI64ArrayAttr(multiples_vals); in matchAndRewrite() local
2086 tfl_tile_op.input(), multiples_attr); in matchAndRewrite()