Searched refs:allowed_dtypes (Results 1 – 5 of 5) sorted by relevance
151 allowed_dtypes = [162 if dtype not in allowed_dtypes:165 % (allowed_dtypes, dtype))
164 allowed_dtypes = [172 if dtype not in allowed_dtypes:175 % (allowed_dtypes, dtype))
170 allowed_dtypes = [179 if dtype not in allowed_dtypes:182 % (allowed_dtypes, dtype))
144 allowed_dtypes = [dtypes.float32, dtypes.complex64]145 if spectrum.dtype not in allowed_dtypes:147 (allowed_dtypes, spectrum.dtype))
1456 const std::set<DataType>& allowed_dtypes, in AllowDataTypes() argument1465 if (!allowed_dtypes.count(op_dtype)) { in AllowDataTypes()1468 for (auto it = allowed_dtypes.begin(); it != allowed_dtypes.end(); ++it) { in AllowDataTypes()1469 if (it != allowed_dtypes.begin()) ss << ", "; in AllowDataTypes()