Searched refs:tf_dtypes (Results 1 – 1 of 1) sorted by relevance
35 from tensorflow.python.framework import dtypes as tf_dtypes unknown71 def _generate_random_tensor_ops(shape: Sequence[int], dtype: tf_dtypes.DType,75 random_dtype = tf_dtypes.float32 if dtype.is_floating else tf_dtypes.int3277 dtype_max = 1 if dtype == tf_dtypes.bool else dtype.max92 dtype = tf_dtypes.as_dtype(tensor_info.dtype)