Home
last modified time | relevance | path

Searched refs:np_dtypes (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_dtypes_test.py26 from tensorflow.python.ops.numpy_ops import np_dtypes
34 np_dtypes.set_allow_float64(False)
35 np_dtypes.set_prefer_float32(prefer_f32)
36 self.assertEqual(dtypes.float32, np_dtypes.default_float_type())
38 np_dtypes._result_type(np.zeros([], np.float64), 1.1))
41 np_dtypes.set_allow_float64(True)
42 np_dtypes.set_prefer_float32(False)
43 self.assertEqual(dtypes.float64, np_dtypes.default_float_type())
44 self.assertEqual(dtypes.float64, np_dtypes._result_type(1.1))
47 np_dtypes.set_allow_float64(True)
[all …]
Dnp_random.py29 from tensorflow.python.ops.numpy_ops import np_dtypes
75 dtype = np_dtypes.default_float_type()
81 dtype = np_dtypes.default_float_type()
96 return random_ops.random_poisson(shape=size, lam=lam, dtype=np_dtypes.int_)
Dnp_config.py22 from tensorflow.python.ops.numpy_ops import np_dtypes
39 np_dtypes.set_prefer_float32(prefer_float32)
Dnp_arrays.py27 from tensorflow.python.ops.numpy_ops import np_dtypes
50 dtype = np_dtypes.default_float_type()
Dnp_random_test.py30 from tensorflow.python.ops.numpy_ops import np_dtypes
50 old_allow_float64 = np_dtypes.is_allow_float64()
51 np_dtypes.set_allow_float64(allow_float64)
63 np_dtypes.set_allow_float64(old_allow_float64)
195 np.float64 if np_dtypes.is_allow_float64() else np.float32)
Dnp_utils.py35 from tensorflow.python.ops.numpy_ops import np_dtypes
498 return np_dtypes._result_type(*arrays_and_dtypes) # pylint: disable=protected-access
504 return np_dtypes._result_type(_maybe_get_dtype(t1), # pylint: disable=protected-access
514 return np_dtypes.canonicalize_dtype(np.promote_types(type1, type2))
D__init__.py178 from tensorflow.python.ops.numpy_ops.np_dtypes import *
Dnp_math_ops.py43 from tensorflow.python.ops.numpy_ops import np_dtypes
121 float_ = np_dtypes.default_float_type()
124 if not np_dtypes.is_allow_float64():
344 if a.dtype == np_dtypes.complex128 or a.dtype == np_dtypes.complex64:
379 y = y.astype(np_dtypes.default_float_type())
600 x = x.astype(np_dtypes.default_float_type())
1240 np_utils.result_type(a.dtype, np_dtypes.default_float_type()))
1253 np_dtypes.default_float_type())
DBUILD17 "np_dtypes.py",
Dnp_array_ops.py41 from tensorflow.python.ops.numpy_ops import np_dtypes
63 np_utils.result_type(dtype) if dtype else np_dtypes.default_float_type())
539 a = math_ops.cast(a, np_dtypes.default_float_type())
735 float_dtype = np_dtypes.default_float_type()
1180 dtype = np_dtypes.default_float_type()
/external/tensorflow/tensorflow/python/ops/
Dmath_ops.py109 np_dtypes = LazyLoader( variable
1173 result_type = np_dtypes._result_type(