Searched refs:tensor_ty (Results 1 – 5 of 5) sorted by relevance
402 TensorType tensor_ty; in ParseTypeWithSubtype() local403 if (parser.parseType(tensor_ty)) return Type(); in ParseTypeWithSubtype()404 subtypes.push_back(tensor_ty); in ParseTypeWithSubtype()
88 if (auto tensor_ty = type.dyn_cast<TensorType>()) in IsValidTFTensorType() local89 return IsValidTFElementType(tensor_ty.getElementType()); in IsValidTFTensorType()
580 auto tensor_ty = tensor.getType().cast<TensorType>(); in GetReshapeOutputType() local581 auto element_ty = tensor_ty.getElementType(); in GetReshapeOutputType()632 if (!tensor_ty.hasStaticShape()) { in GetReshapeOutputType()642 for (const auto &dim : tensor_ty.getShape()) { in GetReshapeOutputType()679 auto tensor_ty = op.tensor().getType().cast<TensorType>(); in Verify() local680 if (output_ty.hasStaticShape() && tensor_ty.hasStaticShape()) { in Verify()682 const int64_t tensor_ty_size = tensor_ty.getNumElements(); in Verify()1929 auto tensor_ty = shape_op.input().getType().dyn_cast<RankedTensorType>(); in fold() local1931 if (!tensor_ty) return {}; in fold()1933 int64_t rank = tensor_ty.getRank(); in fold()[all …]
627 if (auto tensor_ty = type.dyn_cast<TFRTensorType>()) { in printType() local628 attrs = tensor_ty.getAttrKeys(); in printType()
3923 auto tensor_ty = op.tensor().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local3927 if (!tensor_ty || !indices_ty || !updates_ty) return failure(); in matchAndRewrite()3934 int64_t tensor_rank = tensor_ty.getRank(); in matchAndRewrite()3957 RankedTensorType::get(/*shape=*/{}, tensor_ty.getElementType()); in matchAndRewrite()5502 auto tensor_ty = tensor.getType().cast<ShapedType>(); in matchAndRewrite() local5506 if (!result_ty.hasRank() || !tensor_ty.hasRank() || !shape_ty.hasRank()) { in matchAndRewrite()