Searched refs:tensor_ty (Results 1 – 5 of 5) sorted by relevance
60 if (auto tensor_ty = type.dyn_cast<TensorType>()) in IsValidTFTensorType() local61 return IsValidTFElementType(tensor_ty.getElementType()); in IsValidTFTensorType()
84 TensorType tensor_ty; in ParseTypeWithSubtype() local85 if (parser.parseType(tensor_ty)) return Type(); in ParseTypeWithSubtype()89 if (!IsValidTFTensorType(tensor_ty)) { in ParseTypeWithSubtype()90 parser.emitError(parser.getNameLoc()) << "invalid subtype: " << tensor_ty; in ParseTypeWithSubtype()93 subtypes.push_back(tensor_ty); in ParseTypeWithSubtype()
750 auto tensor_ty = tensor.getType().cast<TensorType>(); in GetReshapeOutputType() local751 auto element_ty = tensor_ty.getElementType(); in GetReshapeOutputType()802 if (!tensor_ty.hasStaticShape()) { in GetReshapeOutputType()812 for (const auto &dim : tensor_ty.getShape()) { in GetReshapeOutputType()850 auto tensor_ty = op.tensor().getType().cast<TensorType>(); in verify() local851 if (output_ty.hasStaticShape() && tensor_ty.hasStaticShape()) { in verify()853 const int64_t tensor_ty_size = tensor_ty.getNumElements(); in verify()2069 auto tensor_ty = shape_op.input().getType().dyn_cast<RankedTensorType>(); in fold() local2071 if (!tensor_ty) return {}; in fold()2073 int64_t rank = tensor_ty.getRank(); in fold()[all …]
978 if (auto tensor_ty = type.dyn_cast<TFRTensorType>()) { in printType() local979 attrs = tensor_ty.getAttrKeys(); in printType()
4450 auto tensor_ty = in matchAndRewrite() local4457 if (!tensor_ty || !indices_ty || !updates_ty) return failure(); in matchAndRewrite()4471 if (!tensor_ty.hasStaticShape()) { in matchAndRewrite()4479 auto tensor_shape = tensor_ty.getShape(); in matchAndRewrite()4510 int64_t tensor_rank = tensor_ty.getRank(); in matchAndRewrite()4528 Derived::BuildScatterBody(tensor_ty.getElementType(), in matchAndRewrite()5380 if (auto tensor_ty = ty.dyn_cast<RankedTensorType>()) { in matchAndRewrite() local5381 if (!tensor_ty.hasStaticShape()) return failure(); in matchAndRewrite()