Home
last modified time | relevance | path

Searched refs:begin_attr (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dunroll_batch_matmul.cc103 auto begin_attr = in sliceInput() local
106 auto begin = rewriter.create<TF::ConstOp>(loc, vector3_type, begin_attr); in sliceInput()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_tf.cc705 auto begin_attr = DenseElementsAttr::get<int32_t>(type, padded_begin); in RewriteEllipsisMask() local
706 auto begin_op = rewriter.create<ConstantOp>(op->getLoc(), type, begin_attr); in RewriteEllipsisMask()
765 auto begin_attr = strided_slice_op.begin(); in matchAndRewrite() local
769 auto begin_attr_type = begin_attr.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()
779 !matchPattern(begin_attr, m_Constant(&begin_elem_attr))) { in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc1915 DenseIntElementsAttr begin_attr, end_attr, strides_attr; in fold() local
1916 if (!matchPattern(begin(), m_Constant(&begin_attr)) || in fold()
1919 begin_attr.getNumElements() != 1 || end_attr.getNumElements() != 1 || in fold()
1934 int64_t begin_int = begin_attr.getValue<APInt>(0).getSExtValue(); in fold()