Searched refs:getSubtypes (Results 1 – 16 of 16) sorted by relevance
48 if (resource_type.getSubtypes().size() == 1) in GetResourceSubtypeOrDefault()49 return resource_type.getSubtypes().front(); in GetResourceSubtypeOrDefault()59 .getSubtypes() in HasResourceSubtype()68 .getSubtypes() in GetResourceSubtype()
49 .getSubtypes())) in AllResourceTypesHaveSubtypes()59 .getSubtypes() in GetResourceSubtype()
125 if (resource_type.getSubtypes().size() != 1) in ValidateResourceArgument()204 Type arg_type = resource_type.getSubtypes().front(); in PromoteResourcesToArguments()
195 arg_type.getElementType().cast<TF::ResourceType>().getSubtypes(); in LiftVariables()
220 .getSubtypes()[0]}, in ReadLocalVariable()
192 .getSubtypes(); in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
555 if (!variant_type || variant_type.getSubtypes().size() != 1) return failure(); in GetElementShapeFromResultType()556 TensorType tensor_type = variant_type.getSubtypes().front(); in GetElementShapeFromResultType()
369 .getSubtypes()[0] in HandleTensorArraySizeV3Op()387 .getSubtypes()[0] in CreateAndInitializeGradVariable()
156 auto subtypes = resource_type.getSubtypes(); in GetResourceSubtype()
98 if (!type_with_subtypes || type_with_subtypes.getSubtypes().empty()) { in GetSubtypesHelper()103 for (auto subtype : type_with_subtypes.getSubtypes()) { in GetSubtypesHelper()
193 return variant_type.getSubtypes(); in GetSubtypes()195 return resource_type.getSubtypes(); in GetSubtypes()
228 ArrayRef<TensorType> getSubtypes() { return Base::getImpl()->subtypes_; } in getSubtypes() function
415 ArrayRef<TensorType> subtypes = ty.getSubtypes(); in PrintTypeWithSubtype()
2198 if (variant_type.getSubtypes().empty()) return {}; in fold()2199 return ConvertShapeToAttr(variant_type.getSubtypes()[0], width); in fold()2809 auto subtypes = resource_type.getSubtypes(); in Verify()2813 op, resource_type.getSubtypes().front(), op.getType()); in Verify()2827 if (resource_type.getSubtypes().empty()) return {}; in fold()2828 return ConvertShapeToAttr(resource_type.getSubtypes()[0], width); in fold()
50 if (handle_dtype().getSubtypes().size() != 1) {63 TensorType element_type() { return handle_dtype().getSubtypes()[0]; }876 if (resource_type().getSubtypes().size() != 1) {897 TensorType resource_subtype() { return resource_type().getSubtypes()[0]; }
128 if (handle_dtype.getSubtypes().empty()) { in GetTensorTypeForTensorList()131 return PrependLeadingDimIfRanked(-1, handle_dtype.getSubtypes()[0], rewriter); in GetTensorTypeForTensorList()200 ArrayRef<TensorType> subtypes = variant_ty.getSubtypes(); in matchAndRewrite()