Searched refs:y_dtype (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_matmul_op_test.py | 51 y_dtype=dtypes.float32): argument
54 tf_y = math_ops.cast(y, y_dtype)
80 for y_dtype in (dtypes.float32, dtypes.bfloat16):
81 self._testCpuMatmul(x, y, x_dtype=x_dtype, y_dtype=y_dtype)
88 for y_dtype in (dtypes.float32, dtypes.bfloat16):
89 self._testCpuMatmul(x, y, x_dtype=x_dtype, y_dtype=y_dtype)
96 for y_dtype in (dtypes.float32, dtypes.bfloat16):
97 self._testCpuMatmul(x, y, x_dtype=x_dtype, y_dtype=y_dtype)
107 for y_dtype in (dtypes.float32, dtypes.bfloat16):
110 self._testCpuMatmul(x, y, x_dtype=x_dtype, y_dtype=y_dtype)
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradient_checker_v2.py | 126 def _compute_theoretical_jacobian(f, y_shape, y_dtype, xs, param): argument
149 y_factor = 2 if y_dtype.is_complex else 1
163 dy_data = np.zeros(y_shape, dtype=y_dtype.as_numpy_dtype)
164 dy_data_flat = dy_data.ravel().view(y_dtype.real_dtype.as_numpy_dtype)
167 [y_dtype] + [x.dtype for x in xs])
195 def _compute_numeric_jacobian(f, y_size, y_dtype, xs, param, argument
219 if y_dtype == dtypes.bfloat16:
221 y_dtype = dtypes.float32
225 y_size = y_size * (2 if y_dtype.is_complex else 1)
227 y_dtype = y_dtype.real_dtype.as_numpy_dtype
[all …]
|
| D | math_ops.py | 941 y_dtype = y.dtype.base_dtype
942 if x_dtype != y_dtype:
944 (x_dtype, y_dtype))
961 y_dtype = y.dtype.base_dtype
962 if x_dtype != y_dtype:
964 (x_dtype, y_dtype))
990 y_dtype = y.dtype.base_dtype
991 if x_dtype != y_dtype:
993 (x_dtype, y_dtype))
1076 y_dtype = y.dtype.base_dtype
[all …]
|
| D | gradient_checker.py | 171 y_dtype = y.dtype.real_dtype.as_numpy_dtype
175 scale = np.asarray(2 * delta, dtype=y_dtype)[()]
189 jacobian[row, :] = diff.ravel().view(y_dtype)
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/ |
| D | data_feeder.py | 349 y_dtype = (
351 self._y = (None if y is None else check_array(y, dtype=y_dtype))
|