Searched defs:begin_type (Results 1 – 3 of 3) sorted by relevance
637 RankedTensorType begin_type = begin.getType().cast<RankedTensorType>(); in RewriteEllipsisMask() local
1539 auto begin_type = op.begin().getType().cast<ShapedType>(); in Verify() local1633 auto begin_type = begin.getType().dyn_cast_or_null<RankedTensorType>(); in matchAndRewrite() local
2247 const auto begin_type = context->tensors[node->inputs->data[1]].type; in Validate() local