| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | where_op_test.py | 174 self._testWrongNumbers(array_ops.where_v2)
177 self._testBasicVec(array_ops.where_v2)
180 self._testRandomVec(array_ops.where_v2)
183 self._testBasicMat(array_ops.where_v2)
186 self._testBasic3Tensor(array_ops.where_v2)
189 self._testRandom(np.bool, None, array_ops.where_v2)
192 self._testRandom(np.int32, None, array_ops.where_v2)
195 self._testRandom(np.int64, None, array_ops.where_v2)
198 self._testRandom(np.float32, None, array_ops.where_v2)
201 self._testRandom(np.float64, None, array_ops.where_v2)
[all …]
|
| D | cwise_ops_test.py | 418 self._testScalar(array_ops.where_v2)
437 self._testScalarBroadcast(array_ops.where_v2, c, x, y)
438 self._testScalarBroadcast(array_ops.where_v2, c, y, x)
441 self._testScalarBroadcast(array_ops.where_v2, c, x, y)
442 self._testScalarBroadcast(array_ops.where_v2, c, y, x)
445 self._testScalarBroadcast(array_ops.where_v2, c, x, y)
446 self._testScalarBroadcast(array_ops.where_v2, c, y, x)
449 self._testScalarBroadcast(array_ops.where_v2, c, x, y)
450 self._testScalarBroadcast(array_ops.where_v2, c, y, x)
453 self._testScalarBroadcast(array_ops.where_v2, c, x, y)
[all …]
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | special_math.py | 152 y = array_ops.where_v2(
154 array_ops.where_v2(
240 maybe_complement_p = array_ops.where_v2(p > -np.expm1(-2.), 1. - p, p)
244 sanitized_mcp = array_ops.where_v2(
270 x = array_ops.where_v2(
272 array_ops.where_v2(z >= 8.0, x_for_small_p, x_otherwise))
274 x = array_ops.where_v2(p > 1. - np.exp(-2.), x, -x)
277 x_nan_replaced = array_ops.where_v2(p <= 0.0, -infinity,
278 array_ops.where_v2(p >= 1.0, infinity, x))
365 return array_ops.where_v2(
[all …]
|
| D | uniform.py | 180 return array_ops.where_v2(
182 array_ops.where_v2(
194 result_if_not_big = array_ops.where_v2(
196 return array_ops.where_v2(x >= self.high, ones, result_if_not_big)
|
| D | student_t.py | 287 return array_ops.where_v2(math_ops.less(y, 0.), neg_cdf, 1. - neg_cdf)
310 return array_ops.where_v2(
337 denom = array_ops.where_v2(
345 result_where_defined = array_ops.where_v2(
351 return array_ops.where_v2(
|
| D | util.py | 651 shift = array_ops.where_v2(
703 [array_ops.where_v2(cond, 0, n)], [array_ops.where(cond, n, -1)])
1129 max_log_absw_x = array_ops.where_v2(
1195 x = array_ops.where_v2(
1199 return array_ops.where_v2(
1201 array_ops.where_v2(is_too_large, too_large_value, y))
1337 axis = array_ops.where_v2(axis < 0, ndims + axis, axis)
|
| D | dirichlet.py | 300 return array_ops.where_v2(
|
| D | gamma.py | 273 return array_ops.where_v2(self.concentration > 1., mode, nan)
|
| D | beta.py | 319 return array_ops.where_v2(is_defined, mode, nan)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | ternary_ops_test.py | 120 array_ops.where_v2,
127 array_ops.where_v2,
134 array_ops.where_v2,
141 array_ops.where_v2,
149 array_ops.where_v2,
157 array_ops.where_v2,
165 array_ops.where_v2,
173 array_ops.where_v2,
179 array_ops.where_v2,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops_test.py | 36 array_ops.reshape(array_ops.where_v2(x[0, :, 0] > 0.5), [-1]),
59 array_ops.reshape(array_ops.where_v2(x[0, :, 0] > 0.5), [-1]),
|
| D | math_grad.py | 402 return array_ops.where_v2(is_selected, gathered_grads, zeros), None
441 grad = array_ops.where_v2(
444 non_zero_data = array_ops.where_v2(is_zero, array_ops.ones_like(data), data)
451 partial_derivative = array_ops.where_v2(is_zero, gathered_non_zero_prod,
488 return (array_ops.where_v2(is_positive, gathered,
507 return array_ops.where_v2(is_selected, gathered_grads, zeros), None, None
552 grad = array_ops.where_v2(
555 non_zero_data = array_ops.where_v2(is_zero, array_ops.ones_like(op.inputs[0]),
569 partial_derivative = array_ops.where_v2(is_zero, gathered_non_zero_prod,
952 dy_dx = array_ops.where_v2(
[all …]
|
| D | sparse_ops.py | 132 indices = array_ops.where_v2(
222 axis = array_ops.where_v2(axis >= 0, axis, axis + rank + 1)
1986 where_true = array_ops.reshape(array_ops.where_v2(to_retain), [-1])
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_where_op.py | 31 def where_v2(condition, x=None, y=None, name=None): function
219 return array_ops.where_v2(condition, x, y)
221 return ragged_functional_ops.map_flat_values(array_ops.where_v2, condition, x,
|
| D | ragged_where_op_test.py | 356 result = ragged_where_op.where_v2(condition, x, y)
375 self.evaluate(ragged_where_op.where_v2(condition, x, y))
|
| D | ragged_map_flat_values_op_test.py | 79 op=array_ops.where_v2,
|
| D | ragged_dispatch.py | 502 (array_ops.where_v2, ragged_where_op.where_v2, ['condition', 'x', 'y']),
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_math_ops.py | 373 return array_ops.where_v2(
375 array_ops.where_v2(x1 > 0, constant_op.constant(1, dtype=x2.dtype), x2))
500 x2_safe = array_ops.where_v2(x2 != 0, x2, constant_op.constant(1, x2.dtype))
501 x1, x2 = (array_ops.where_v2(x2 != 0, x2, x1),
502 array_ops.where_v2(x2 != 0, math_ops.mod(x1, x2_safe),
504 return (array_ops.where_v2(x1 < x2, x2,
505 x1), array_ops.where_v2(x1 < x2, x1, x2))
532 d_safe = array_ops.where_v2(
534 return array_ops.where_v2(
755 return array_ops.where_v2(x < 0, np.pi, 0)
[all …]
|
| D | np_array_ops.py | 806 x = array_ops.where_v2(x < 0, np_utils.add(x, a_rank), x)
940 return array_ops.where_v2(condition, x, y)
1156 array_ops.where_v2(math_ops.cast(a, dtypes.bool)),
1218 return array_ops.where_v2(
1239 return array_ops.where_v2(
1430 indices = array_ops.where_v2(indices < 0, indices + arr_shape[axis], indices)
1481 data = array_ops.where_v2(data)
1697 stacked_indices = array_ops.where_v2(
1755 stacked_indices = array_ops.where_v2(stacked_indices < 0,
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | random_seed.py | 52 seed2 = array_ops.where_v2(
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | where.py | 79 where = tf.where_v2 if parameters["use_where_v2"] else tf.where
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linalg_impl.py | 285 return array_ops.where_v2(
288 return array_ops.where_v2(
339 return i + 1, array_ops.where_v2(
898 singular_values = array_ops.where_v2(
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | math_test.py | 619 return array_ops.where_v2(cond_i, a_i, b_i)
639 return array_ops.where_v2(cond_i, a_i, b_i)
658 return array_ops.where_v2(cond, a_i, b_i)
|
| D | xla_control_flow_ops_test.py | 141 array_ops.where_v2(not_done, x, y) for x, y in zip(outputs, args))
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 4404 output = array_ops.where_v2(tiled_mask_t, output, prev_output)
4410 array_ops.where_v2(m, s, ps)
4421 last_output = array_ops.where_v2(
4424 outputs = array_ops.where_v2(
4506 array_ops.where_v2(m, o, fm)
4697 tile_shape = array_ops.where_v2(shape_diff > 0, expr_shape,
4700 x = array_ops.where_v2(condition, then_expression, else_expression)
4850 return array_ops.where_v2(x > 0, res, alpha * res)
6222 return array_ops.where_v2(
|