Home
last modified time | relevance | path

Searched refs:float_dtypes (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client_test.py52 float_dtypes = [bfloat16, np.float32, np.float64]
54 standard_dtypes = int_dtypes + float_dtypes + complex_dtypes + [np.bool_]
57 float_dtypes = [np.float32]
59 standard_dtypes = int_dtypes + float_dtypes + complex_dtypes + [np.bool_]
60 dlpack_dtypes = int_dtypes + float_dtypes + [np.bool_]
160 } for dtype in int_dtypes + float_dtypes)
171 } for dtype in float_dtypes)
183 } for dtype in float_dtypes)
195 } for dtype in float_dtypes)
245 } for dtype in float_dtypes)
[all …]
/external/tensorflow/tensorflow/python/dlpack/
Ddlpack_test.py36 float_dtypes = [np.float16, np.float32, np.float64] variable
38 dlpack_dtypes = int_dtypes + float_dtypes + [dtypes.bfloat16]