Searched defs:begin_type (Results 1 – 5 of 5) sorted by relevance
378 auto begin_type = begin.getType().dyn_cast_or_null<RankedTensorType>(); in matchAndRewrite() local
547 RankedTensorType begin_type = begin.getType().cast<RankedTensorType>(); in RewriteEllipsisMask() local
2011 auto begin_type = op.begin().getType().cast<ShapedType>(); in verify() local2103 auto begin_type = begin.getType().dyn_cast_or_null<RankedTensorType>(); in matchAndRewrite() local3186 if (auto begin_type = op.begin().getType().dyn_cast<RankedTensorType>()) { in verify() local
2909 const auto begin_type = context->tensors[node->inputs->data[1]].type; in Validate() local
3085 auto begin_type = begin_indices.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local