Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/python/keras/engine/
Dcompile_utils_test.py58 sw = ops.convert_to_tensor_v2_with_dispatch([0, 0, 0, 0, 0, 1, 1, 1, 1, 1])
91 sw = ops.convert_to_tensor_v2_with_dispatch([0, 0, 0, 0, 0, 1, 1, 1, 1, 1])
117 sw = ops.convert_to_tensor_v2_with_dispatch([0, 0, 0, 0, 0, 1, 1, 1, 1, 1])
140 sw = ops.convert_to_tensor_v2_with_dispatch([0, 0, 0, 0, 0, 1, 1, 1, 1, 1])
175 sw = ops.convert_to_tensor_v2_with_dispatch([0, 0, 0, 0, 0, 1, 1, 1, 1, 1])
198 sw = ops.convert_to_tensor_v2_with_dispatch([0, 0, 0, 0, 0, 1, 1, 1, 1, 1])
225 'output1': ops.convert_to_tensor_v2_with_dispatch([[0], [1], [2]]),
226 'output2': ops.convert_to_tensor_v2_with_dispatch([[3], [4], [5]]),
227 'output3': ops.convert_to_tensor_v2_with_dispatch([[6], [7], [8]])
230 'output1': ops.convert_to_tensor_v2_with_dispatch([[1], [2], [3]]),
[all …]
Ddata_adapter_test.py455 ops.convert_to_tensor_v2_with_dispatch(self.arraylike_input)
927 yield (ops.convert_to_tensor_v2_with_dispatch([step]),)
960 yield (ops.convert_to_tensor_v2_with_dispatch([step]),)
1039 x = ops.convert_to_tensor_v2_with_dispatch([0, 1, 2, 3, 4])
1040 y = ops.convert_to_tensor_v2_with_dispatch([0, 2, 4, 6, 8])
1041 sw = ops.convert_to_tensor_v2_with_dispatch([0, 4, 8, 12, 16])
1047 train_x = ops.convert_to_tensor_v2_with_dispatch(train_x)
1048 train_y = ops.convert_to_tensor_v2_with_dispatch(train_y)
1049 train_sw = ops.convert_to_tensor_v2_with_dispatch(train_sw)
1050 val_x = ops.convert_to_tensor_v2_with_dispatch(val_x)
[all …]
Dtraining_eager_v1.py125 inputs = nest.map_structure(ops.convert_to_tensor_v2_with_dispatch, inputs)
137 ops.convert_to_tensor_v2_with_dispatch(val))
Dbase_preprocessing_layer_v1.py104 data = ops.convert_to_tensor_v2_with_dispatch(data)
/external/tensorflow/tensorflow/python/keras/layers/
Dsubclassed_layers_test.py40 self.b = ops.convert_to_tensor_v2_with_dispatch(2.0)
49 x = ops.convert_to_tensor_v2_with_dispatch([[3.0]])
61 a = ops.convert_to_tensor_v2_with_dispatch(1.0)
64 self.constant = ops.convert_to_tensor_v2_with_dispatch(self.variable)
73 x = ops.convert_to_tensor_v2_with_dispatch([[3.0]])
Drnn_cell_wrapper_v2_test.py43 x = ops.convert_to_tensor_v2_with_dispatch(
45 m = ops.convert_to_tensor_v2_with_dispatch(
67 x = ops.convert_to_tensor_v2_with_dispatch(
69 m = ops.convert_to_tensor_v2_with_dispatch(
122 inputs = ops.convert_to_tensor_v2_with_dispatch([[[1]]],
147 inputs = ops.convert_to_tensor_v2_with_dispatch([[1]], dtype=dtypes.float32)
148 state = ops.convert_to_tensor_v2_with_dispatch([[1]], dtype=dtypes.float32)
/external/tensorflow/tensorflow/python/keras/utils/
Dtf_utils_test.py51 ops.convert_to_tensor_v2_with_dispatch(0.)))
60 ops.convert_to_tensor_v2_with_dispatch(0.)))
70 return ops.convert_to_tensor_v2_with_dispatch(42.)
82 ops.convert_to_tensor_v2_with_dispatch(0.)))
92 ops.convert_to_tensor_v2_with_dispatch(0.)))
107 self.value = ops.convert_to_tensor_v2_with_dispatch([[42.]])
122 x = ops.convert_to_tensor_v2_with_dispatch(d)
150 y = model(ops.convert_to_tensor_v2_with_dispatch([[7.]]))
Dlosses_utils.py111 predictions = ops.convert_to_tensor_v2_with_dispatch(predictions)
113 labels = ops.convert_to_tensor_v2_with_dispatch(labels)
304 losses = ops.convert_to_tensor_v2_with_dispatch(losses)
308 sample_weight = ops.convert_to_tensor_v2_with_dispatch(sample_weight)
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_util.py117 return ops.convert_to_tensor_v2_with_dispatch(
192 x = ops.convert_to_tensor_v2_with_dispatch(x, name="x")
195 zero = ops.convert_to_tensor_v2_with_dispatch(0, dtype=dtype.real_dtype)
211 x = ops.convert_to_tensor_v2_with_dispatch(x, name="x")
217 zero = ops.convert_to_tensor_v2_with_dispatch(0, dtype=dtype.real_dtype)
264 return ops.convert_to_tensor_v2_with_dispatch(shape, dtype=dtype, name=name)
326 batch_matrices[i] = ops.convert_to_tensor_v2_with_dispatch(mat)
369 matrix = ops.convert_to_tensor_v2_with_dispatch(matrix, name="matrix")
370 rhs = ops.convert_to_tensor_v2_with_dispatch(
530 arg_dims = [ops.convert_to_tensor_v2_with_dispatch(
Dlinear_operator_block_diag.py275 return ops.convert_to_tensor_v2_with_dispatch(
342 block = ops.convert_to_tensor_v2_with_dispatch(block)
347 x = ops.convert_to_tensor_v2_with_dispatch(x, name="x")
416 block = ops.convert_to_tensor_v2_with_dispatch(block)
424 x = ops.convert_to_tensor_v2_with_dispatch(x, name="x")
520 block = ops.convert_to_tensor_v2_with_dispatch(block)
525 rhs = ops.convert_to_tensor_v2_with_dispatch(rhs, name="rhs")
595 block = ops.convert_to_tensor_v2_with_dispatch(block)
603 rhs = ops.convert_to_tensor_v2_with_dispatch(rhs, name="rhs")
Dlinear_operator_block_lower_triangular.py379 return ops.convert_to_tensor_v2_with_dispatch(
446 block = ops.convert_to_tensor_v2_with_dispatch(block)
451 x = ops.convert_to_tensor_v2_with_dispatch(x, name="x")
556 block = ops.convert_to_tensor_v2_with_dispatch(block)
564 x = ops.convert_to_tensor_v2_with_dispatch(x, name="x")
687 block = ops.convert_to_tensor_v2_with_dispatch(block)
697 rhs = ops.convert_to_tensor_v2_with_dispatch(rhs, name="rhs")
808 block = ops.convert_to_tensor_v2_with_dispatch(block)
815 rhs = ops.convert_to_tensor_v2_with_dispatch(rhs, name="rhs")
Dlinear_operator_toeplitz.py222 col = ops.convert_to_tensor_v2_with_dispatch(self.col)
223 row = ops.convert_to_tensor_v2_with_dispatch(self.row)
249 row = ops.convert_to_tensor_v2_with_dispatch(self.row)
250 col = ops.convert_to_tensor_v2_with_dispatch(self.col)
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)
Dlinear_operator.py419 return ops.convert_to_tensor_v2_with_dispatch(self.tensor_rank)
463 return ops.convert_to_tensor_v2_with_dispatch(dim_value)
507 return ops.convert_to_tensor_v2_with_dispatch(dim_value)
675 x = ops.convert_to_tensor_v2_with_dispatch(x, name="x")
722 x = ops.convert_to_tensor_v2_with_dispatch(x, name="x")
868 rhs = ops.convert_to_tensor_v2_with_dispatch(rhs, name="rhs")
925 rhs = ops.convert_to_tensor_v2_with_dispatch(rhs, name="rhs")
1088 x = ops.convert_to_tensor_v2_with_dispatch(x, name="x")
Dlinear_operator_householder.py209 reflection_axis = ops.convert_to_tensor_v2_with_dispatch(
241 reflection_axis = ops.convert_to_tensor_v2_with_dispatch(
251 reflection_axis = ops.convert_to_tensor_v2_with_dispatch(
Dlinear_operator_tridiag.py259 diagonals = ops.convert_to_tensor_v2_with_dispatch(self.diagonals)
367 ops.convert_to_tensor_v2_with_dispatch(d) for d in self.diagonals
Dlinear_operator_full_matrix.py171 matrix = ops.convert_to_tensor_v2_with_dispatch(matrix, name="matrix")
/external/tensorflow/tensorflow/python/keras/
Dlosses.py1219 y_pred = ops.convert_to_tensor_v2_with_dispatch(y_pred)
1318 y_pred = ops.convert_to_tensor_v2_with_dispatch(y_pred)
1357 y_pred = ops.convert_to_tensor_v2_with_dispatch(y_pred)
1394 y_pred = ops.convert_to_tensor_v2_with_dispatch(y_pred)
1442 y_pred = ops.convert_to_tensor_v2_with_dispatch(y_pred)
1475 y_pred = ops.convert_to_tensor_v2_with_dispatch(y_pred)
1508 y_pred = ops.convert_to_tensor_v2_with_dispatch(y_pred)
1543 half = ops.convert_to_tensor_v2_with_dispatch(0.5, dtype=abs_error.dtype)
1581 y_pred = ops.convert_to_tensor_v2_with_dispatch(y_pred)
1620 y_pred = ops.convert_to_tensor_v2_with_dispatch(y_pred)
[all …]
/external/tensorflow/tensorflow/python/keras/distribute/
Dcustom_training_loop_optimizer_test.py59 ops.convert_to_tensor_v2_with_dispatch([1., 1.]),
60 ops.convert_to_tensor_v2_with_dispatch([2., 2.]),
88 grads = ops.convert_to_tensor_v2_with_dispatch([1., 1.])
110 grads = ops.convert_to_tensor_v2_with_dispatch([1., 1.])
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Dlearning_rate_schedule.py146 initial_learning_rate = ops.convert_to_tensor_v2_with_dispatch(
240 boundaries = nest.map_structure(ops.convert_to_tensor_v2_with_dispatch,
242 values = nest.map_structure(ops.convert_to_tensor_v2_with_dispatch,
244 x_recomp = ops.convert_to_tensor_v2_with_dispatch(step)
377 initial_learning_rate = ops.convert_to_tensor_v2_with_dispatch(
497 initial_learning_rate = ops.convert_to_tensor_v2_with_dispatch(
592 initial_learning_rate = ops.convert_to_tensor_v2_with_dispatch(
692 initial_learning_rate = ops.convert_to_tensor_v2_with_dispatch(
829 initial_learning_rate = ops.convert_to_tensor_v2_with_dispatch(
955 initial_learning_rate = ops.convert_to_tensor_v2_with_dispatch(
Dlegacy_learning_rate_decay.py151 boundaries = nest.map_structure(ops.convert_to_tensor_v2_with_dispatch,
153 values = nest.map_structure(ops.convert_to_tensor_v2_with_dispatch,
155 x_recomp = ops.convert_to_tensor_v2_with_dispatch(x)
Dadadelta.py104 epsilon=ops.convert_to_tensor_v2_with_dispatch(
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/
Dtable_utils.py91 keys = ops.convert_to_tensor_v2_with_dispatch(
93 values = ops.convert_to_tensor_v2_with_dispatch(
184 flat_values = ops.convert_to_tensor_v2_with_dispatch(
192 inputs = ops.convert_to_tensor_v2_with_dispatch(inputs)
Dtext_vectorization.py400 data = ops.convert_to_tensor_v2_with_dispatch(data)
552 inputs = ops.convert_to_tensor_v2_with_dispatch(inputs)
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale_optimizer_test.py141 loss = ops.convert_to_tensor_v2_with_dispatch(5.)
144 loss = ops.convert_to_tensor_v2_with_dispatch(5., dtype='float16')
153 ops.convert_to_tensor_v2_with_dispatch(3.), None,
154 ops.convert_to_tensor_v2_with_dispatch(-4., dtype='float16')
165 ops.convert_to_tensor_v2_with_dispatch([[4., 2.], [8., 5.]]),
166 ops.convert_to_tensor_v2_with_dispatch([1, 3], dtype='int32'),
167 dense_shape=ops.convert_to_tensor_v2_with_dispatch([5, 2],

123