Home
last modified time | relevance | path

Searched refs:convert_to_tensor_v2_with_dispatch (Results 1 – 25 of 42) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/python/keras/
H A Dlosses.py1216 y_pred = ops.convert_to_tensor_v2_with_dispatch(y_pred)
1344 y_pred = ops.convert_to_tensor_v2_with_dispatch(y_pred)
1383 y_pred = ops.convert_to_tensor_v2_with_dispatch(y_pred)
1429 y_pred = ops.convert_to_tensor_v2_with_dispatch(y_pred)
1486 y_pred = ops.convert_to_tensor_v2_with_dispatch(y_pred)
1519 y_pred = ops.convert_to_tensor_v2_with_dispatch(y_pred)
1552 y_pred = ops.convert_to_tensor_v2_with_dispatch(y_pred)
1587 half = ops.convert_to_tensor_v2_with_dispatch(0.5, dtype=abs_error.dtype)
1624 y_pred = ops.convert_to_tensor_v2_with_dispatch(y_pred)
1667 y_pred = ops.convert_to_tensor_v2_with_dispatch(y_pred)
[all …]
H A Dmetrics.py1026 return ops.convert_to_tensor_v2_with_dispatch(result)
3445 y_pred = ops.convert_to_tensor_v2_with_dispatch(y_pred)
3503 y_pred = ops.convert_to_tensor_v2_with_dispatch(y_pred)
3504 y_true = ops.convert_to_tensor_v2_with_dispatch(y_true)
3571 y_pred_rank = ops.convert_to_tensor_v2_with_dispatch(y_pred).shape.ndims
3572 y_true_rank = ops.convert_to_tensor_v2_with_dispatch(y_true).shape.ndims
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/linalg/
H A Dlinear_operator_util.py114 return ops.convert_to_tensor_v2_with_dispatch(
189 x = ops.convert_to_tensor_v2_with_dispatch(x, name="x")
192 zero = ops.convert_to_tensor_v2_with_dispatch(0, dtype=dtype.real_dtype)
208 x = ops.convert_to_tensor_v2_with_dispatch(x, name="x")
214 zero = ops.convert_to_tensor_v2_with_dispatch(0, dtype=dtype.real_dtype)
261 return ops.convert_to_tensor_v2_with_dispatch(shape, dtype=dtype, name=name)
323 batch_matrices[i] = ops.convert_to_tensor_v2_with_dispatch(mat)
366 matrix = ops.convert_to_tensor_v2_with_dispatch(matrix, name="matrix")
367 rhs = ops.convert_to_tensor_v2_with_dispatch(
527 arg_dims = [ops.convert_to_tensor_v2_with_dispatch(
H A Dlinear_operator_block_diag.py278 return ops.convert_to_tensor_v2_with_dispatch(
368 block = ops.convert_to_tensor_v2_with_dispatch(block)
373 x = ops.convert_to_tensor_v2_with_dispatch(x, name="x")
444 block = ops.convert_to_tensor_v2_with_dispatch(block)
452 x = ops.convert_to_tensor_v2_with_dispatch(x, name="x")
569 block = ops.convert_to_tensor_v2_with_dispatch(block)
574 rhs = ops.convert_to_tensor_v2_with_dispatch(rhs, name="rhs")
644 block = ops.convert_to_tensor_v2_with_dispatch(block)
652 rhs = ops.convert_to_tensor_v2_with_dispatch(rhs, name="rhs")
H A Dlinear_operator_block_lower_triangular.py384 return ops.convert_to_tensor_v2_with_dispatch(
451 block = ops.convert_to_tensor_v2_with_dispatch(block)
456 x = ops.convert_to_tensor_v2_with_dispatch(x, name="x")
561 block = ops.convert_to_tensor_v2_with_dispatch(block)
569 x = ops.convert_to_tensor_v2_with_dispatch(x, name="x")
692 block = ops.convert_to_tensor_v2_with_dispatch(block)
702 rhs = ops.convert_to_tensor_v2_with_dispatch(rhs, name="rhs")
813 block = ops.convert_to_tensor_v2_with_dispatch(block)
820 rhs = ops.convert_to_tensor_v2_with_dispatch(rhs, name="rhs")
H A Dlinear_operator_toeplitz.py217 col = ops.convert_to_tensor_v2_with_dispatch(self.col)
218 row = ops.convert_to_tensor_v2_with_dispatch(self.row)
244 row = ops.convert_to_tensor_v2_with_dispatch(self.row)
245 col = ops.convert_to_tensor_v2_with_dispatch(self.col)
H A Dlinear_operator.py423 return ops.convert_to_tensor_v2_with_dispatch(self.tensor_rank)
467 return ops.convert_to_tensor_v2_with_dispatch(dim_value)
511 return ops.convert_to_tensor_v2_with_dispatch(dim_value)
679 x = ops.convert_to_tensor_v2_with_dispatch(x, name="x")
726 x = ops.convert_to_tensor_v2_with_dispatch(x, name="x")
872 rhs = ops.convert_to_tensor_v2_with_dispatch(rhs, name="rhs")
929 rhs = ops.convert_to_tensor_v2_with_dispatch(rhs, name="rhs")
1092 x = ops.convert_to_tensor_v2_with_dispatch(x, name="x")
H A Dlinear_operator_circulant.py115 length_scale = ops.convert_to_tensor_v2_with_dispatch(
120 power = ops.convert_to_tensor_v2_with_dispatch(
123 divisor = ops.convert_to_tensor_v2_with_dispatch(
146 zero_inflation = ops.convert_to_tensor_v2_with_dispatch(
507 spectrum = ops.convert_to_tensor_v2_with_dispatch(spectrum, name="spectrum")
542 return ops.convert_to_tensor_v2_with_dispatch(
H A Dlinear_operator_permutation.py210 perm = ops.convert_to_tensor_v2_with_dispatch(self.perm)
245 perm = ops.convert_to_tensor_v2_with_dispatch(self.perm)
251 perm = ops.convert_to_tensor_v2_with_dispatch(self.perm)
H A Dlinear_operator_householder.py205 reflection_axis = ops.convert_to_tensor_v2_with_dispatch(
236 reflection_axis = ops.convert_to_tensor_v2_with_dispatch(
245 reflection_axis = ops.convert_to_tensor_v2_with_dispatch(
H A Dlinear_operator_tridiag.py256 diagonals = ops.convert_to_tensor_v2_with_dispatch(self.diagonals)
364 ops.convert_to_tensor_v2_with_dispatch(d) for d in self.diagonals
H A Dlinear_operator_low_rank_update.py368 u = ops.convert_to_tensor_v2_with_dispatch(self.u)
372 v = ops.convert_to_tensor_v2_with_dispatch(self.v)
H A Dlinear_operator_identity.py406 mat = ops.convert_to_tensor_v2_with_dispatch(mat, name="mat")
760 mat = ops.convert_to_tensor_v2_with_dispatch(mat, name="mat")
H A Dlinear_operator_full_matrix.py166 matrix = ops.convert_to_tensor_v2_with_dispatch(matrix, name="matrix")
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/optimizer_v2/
H A Dlearning_rate_schedule.py178 initial_learning_rate = ops.convert_to_tensor_v2_with_dispatch(
272 boundaries = nest.map_structure(ops.convert_to_tensor_v2_with_dispatch,
274 values = nest.map_structure(ops.convert_to_tensor_v2_with_dispatch,
276 x_recomp = ops.convert_to_tensor_v2_with_dispatch(step)
409 initial_learning_rate = ops.convert_to_tensor_v2_with_dispatch(
529 initial_learning_rate = ops.convert_to_tensor_v2_with_dispatch(
624 initial_learning_rate = ops.convert_to_tensor_v2_with_dispatch(
724 initial_learning_rate = ops.convert_to_tensor_v2_with_dispatch(
861 initial_learning_rate = ops.convert_to_tensor_v2_with_dispatch(
987 initial_learning_rate = ops.convert_to_tensor_v2_with_dispatch(
H A Dlegacy_learning_rate_decay.py148 boundaries = nest.map_structure(ops.convert_to_tensor_v2_with_dispatch,
150 values = nest.map_structure(ops.convert_to_tensor_v2_with_dispatch,
152 x_recomp = ops.convert_to_tensor_v2_with_dispatch(x)
H A Dadam.py145 epsilon=ops.convert_to_tensor_v2_with_dispatch(
398 epsilon=ops.convert_to_tensor_v2_with_dispatch(
H A Dadadelta.py93 epsilon=ops.convert_to_tensor_v2_with_dispatch(
H A Dadagrad.py95 epsilon=ops.convert_to_tensor_v2_with_dispatch(
H A Dadamax.py122 epsilon=ops.convert_to_tensor_v2_with_dispatch(
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/utils/
H A Dlosses_utils.py119 predictions = ops.convert_to_tensor_v2_with_dispatch(predictions)
121 labels = ops.convert_to_tensor_v2_with_dispatch(labels)
313 losses = ops.convert_to_tensor_v2_with_dispatch(losses)
317 sample_weight = ops.convert_to_tensor_v2_with_dispatch(sample_weight)
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/
H A Dtraining_eager_v1.py121 inputs = nest.map_structure(ops.convert_to_tensor_v2_with_dispatch, inputs)
133 ops.convert_to_tensor_v2_with_dispatch(val))
H A Dbase_layer_v1.py697 return ops.convert_to_tensor_v2_with_dispatch(x)
1038 loss = ops.convert_to_tensor_v2_with_dispatch(
1054 loss = ops.convert_to_tensor_v2_with_dispatch(
1216 update = ops.convert_to_tensor_v2_with_dispatch(x)
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/mixed_precision/
H A Dloss_scale_optimizer.py210 return ops.convert_to_tensor_v2_with_dispatch(self._current_loss_scale)
461 return ops.convert_to_tensor_v2_with_dispatch(
464 return ops.convert_to_tensor_v2_with_dispatch(self._loss_scale)
H A Dautocast_variable.py141 return ops.convert_to_tensor_v2_with_dispatch(self._variable, dtype=dtype,
148 val = ops.convert_to_tensor_v2_with_dispatch(

12