Home
last modified time | relevance | path

Searched refs:end_attr (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_tf.cc707 auto end_attr = DenseElementsAttr::get<int32_t>(type, padded_end); in RewriteEllipsisMask() local
708 auto end_op = rewriter.create<ConstantOp>(op->getLoc(), type, end_attr); in RewriteEllipsisMask()
766 auto end_attr = strided_slice_op.end(); in matchAndRewrite() local
770 auto end_attr_type = end_attr.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()
782 if (!end_attr_type || !matchPattern(end_attr, m_Constant(&end_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
1917 !matchPattern(end(), m_Constant(&end_attr)) || in fold()
1919 begin_attr.getNumElements() != 1 || end_attr.getNumElements() != 1 || in fold()
1935 int64_t end_int = end_attr.getValue<APInt>(0).getSExtValue(); in fold()