Searched refs:AllowedDataTypes (Results 1 – 1 of 1) sorted by relevance
824 DataTypeSet AllowedDataTypes(const OpDef::AttrDef& attr_def) { in AllowedDataTypes() function836 DataTypeSet AllowedDataTypes(const OpDef& op_def, const TypeAttrId& t_attr_id) { in AllowedDataTypes() function842 return AllowedDataTypes(*attr_def); in AllowedDataTypes()1200 return AllowedDataTypes(*op_def, node_type.type_attr) in SupportsF16()1211 return AllowedDataTypes(*op_def, node_type.type_attr).Contains(target_dtype_); in SupportsF16DataType()