Searched refs:begin_type (Results 1 – 3 of 3) sorted by relevance
1539 auto begin_type = op.begin().getType().cast<ShapedType>(); in Verify() local1541 if (input_type.hasStaticShape() && begin_type.hasStaticShape() && in Verify()1543 if (input_type.getRank() != begin_type.getNumElements()) { in Verify()1633 auto begin_type = begin.getType().dyn_cast_or_null<RankedTensorType>(); in matchAndRewrite() local1640 if (begin_type == nullptr && size_type == nullptr) return failure(); in matchAndRewrite()1643 if (begin_op && begin_type && begin_type.getElementType().isInteger(64)) { in matchAndRewrite()1645 begin_op, begin_type, slice_op.getLoc(), &rewriter); in matchAndRewrite()
637 RankedTensorType begin_type = begin.getType().cast<RankedTensorType>(); in RewriteEllipsisMask() local638 const ArrayRef<int64_t> begin_shape = begin_type.getShape(); in RewriteEllipsisMask()
2247 const auto begin_type = context->tensors[node->inputs->data[1]].type; in Validate() local2251 Expect(begin_type == kTfLiteInt32, in Validate()