Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_tf.cc630 auto end_attr = DenseElementsAttr::get<int32_t>(type, padded_end); in RewriteEllipsisMask() local
632 rewriter.create<arith::ConstantOp>(op->getLoc(), type, end_attr); in RewriteEllipsisMask()
687 auto end_attr = strided_slice_op.end(); in matchAndRewrite() local
691 auto end_attr_type = end_attr.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()
703 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.cc2055 DenseIntElementsAttr begin_attr, end_attr, strides_attr; in fold() local
2057 !matchPattern(end(), m_Constant(&end_attr)) || in fold()
2059 begin_attr.getNumElements() != 1 || end_attr.getNumElements() != 1 || in fold()
2075 int64_t end_int = end_attr.getValues<APInt>()[0].getSExtValue(); in fold()
/external/tensorflow/tensorflow/core/transforms/constant_folding/
Dpass.cc1630 ElementsAttr end_attr = end_op->getAttrOfType<ElementsAttr>("value"); in matchAndRewrite() local
1668 int e = end_attr.getElementType().isInteger(32) in matchAndRewrite()
1669 ? end_attr.getValues<int32_t>()[j] in matchAndRewrite()
1670 : end_attr.getValues<int64_t>()[j]; in matchAndRewrite()