Home
last modified time | relevance | path

Searched refs:ones_like (Results 1 – 25 of 137) sorted by relevance

123456

/external/tensorflow/tensorflow/python/keras/layers/
Dconvolutional_recurrent_test.py113 out1 = model.predict(np.ones_like(inputs))
116 model.train_on_batch(np.ones_like(inputs),
118 out2 = model.predict(np.ones_like(inputs))
126 out3 = model.predict(np.ones_like(inputs))
131 out4 = model.predict(np.ones_like(inputs))
135 out5 = model.predict(np.ones_like(inputs))
/external/tensorflow/tensorflow/python/eager/
Dtape_test.py44 math_ops.matmul(array_ops.ones_like(b * dr), b, transpose_b=True),
46 math_ops.matmul(a, array_ops.ones_like(a) * dr, transpose_b=True)
97 array_ops.ones_like(aa),
111 array_ops.ones_like(aa),
125 array_ops.ones_like(aa),
/external/tensorflow/tensorflow/contrib/boosted_trees/python/utils/
Dlosses.py137 one_minus_eps = array_ops.ones_like(
233 ones = array_ops.ones_like(logits, dtype=logits.dtype)
234 neg_ones = -array_ops.ones_like(logits, dtype=logits.dtype)
307 ones = array_ops.ones_like(logits, dtype=logits.dtype)
308 neg_ones = -array_ops.ones_like(logits, dtype=logits.dtype)
/external/tensorflow/tensorflow/python/ops/distributions/
Dmultinomial.py251 n_draws = array_ops.ones_like(
253 logits = array_ops.ones_like(
295 p = self.probs * array_ops.ones_like(
303 p = self.probs * array_ops.ones_like(
Dnormal.py223 scale = self.scale * array_ops.ones_like(self.loc)
227 return self.loc * array_ops.ones_like(self.scale)
233 return self.scale * array_ops.ones_like(self.loc)
Dbernoulli.py145 return (array_ops.ones_like(event) * logits,
146 array_ops.ones_like(logits) * event)
/external/tensorflow/tensorflow/contrib/distributions/python/ops/
Dquantized_distribution.py363 ones = array_ops.ones_like(x_samps)
470 neg_inf = -np.inf * array_ops.ones_like(result_so_far)
508 array_ops.ones_like(result_so_far),
541 neg_inf = -np.inf * array_ops.ones_like(result_so_far)
571 array_ops.ones_like(result_so_far),
Dlogistic.py219 scale = self.scale * array_ops.ones_like(self.loc)
223 return self.loc * array_ops.ones_like(self.scale)
226 return self.scale * array_ops.ones_like(self.loc) * math.pi / math.sqrt(3)
Dgeometric.py162 x *= array_ops.ones_like(self.probs)
174 x *= array_ops.ones_like(self.probs)
175 probs = self.probs * array_ops.ones_like(x)
Dgumbel.py218 scale = self.scale * array_ops.ones_like(self.loc)
225 return self.scale * array_ops.ones_like(self.loc) * math.pi / math.sqrt(6)
228 return self.loc * array_ops.ones_like(self.scale)
Dkumaraswamy.py206 expanded_concentration1 = array_ops.ones_like(
208 expanded_concentration0 = array_ops.ones_like(
Dcauchy.py201 return h * array_ops.ones_like(self.loc)
207 return self.loc * array_ops.ones_like(self.scale)
Donehot_categorical.py203 logits = array_ops.ones_like(x, dtype=logits.dtype) * logits
204 x = array_ops.ones_like(logits, dtype=x.dtype) * x
Drelaxed_onehot_categorical.py278 logits = array_ops.ones_like(x, dtype=logits.dtype) * logits
279 x = array_ops.ones_like(logits, dtype=x.dtype) * x
/external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/
Dbeam_search_ops_test.py121 self.assertAllClose(b_value, end_token * np.ones_like(b_value))
134 v[:found - 1] >= 0, np.ones_like(v[:found - 1], dtype=bool))
136 end_token * np.ones_like(v[found + 1:]))
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_math_ops.py279 array_ops.ones_like(data.flat_values), data.nested_row_splits)
293 array_ops.ones_like(data.flat_values), data.nested_row_splits)
535 array_ops.ones_like(input_tensor.flat_values),
538 ones = array_ops.ones_like(input_tensor)
Dragged_batch_gather_with_default_op.py159 array_ops.ones_like(params_shape.partitioned_dim_sizes[-1]),)
182 array_ops.ones_like(gather_dimension),) * (
/external/tensorflow/tensorflow/python/ops/
Dweights_broadcast_ops.py38 (values_shape_2d, array_ops.ones_like(values_shape_2d)), axis=1)
169 weights, array_ops.ones_like(values), name=scope)
/external/tensorflow/tensorflow/contrib/bayesflow/python/kernel_tests/
Dmonte_carlo_test.py197 self.assertAllEqual(np.ones_like(efx_true_grad_), efx_true_grad_)
202 self.assertAllEqual(np.ones_like(efx_true_grad_, dtype=np.bool),
204 self.assertAllEqual(np.ones_like(efx_true_grad_, dtype=np.bool),
/external/tensorflow/tensorflow/contrib/image/python/ops/
Dinterpolate_spline.py131 ones = array_ops.ones_like(c[..., :1], dtype=c.dtype)
185 array_ops.ones_like(query_points[..., :1], train_points.dtype)
/external/tensorflow/tensorflow/contrib/labeled_tensor/
D__init__.py118 ones_like = _ops.ones_like variable
/external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/python/ops/
Dtraining_ops.py109 array_ops.expand_dims(array_ops.ones_like(tree_thresholds_tensor), 0), 2)
266 array_ops.expand_dims(array_ops.ones_like(tree_thresholds_tensor), 0), 2)
/external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/
Dstatistical_testing_test.py53 np.ones_like(below_threshold, np.bool), below_threshold,
91 np.ones_like(below_threshold, np.bool), below_threshold,
/external/tensorflow/tensorflow/contrib/eager/python/examples/gan/
Dmnist.py166 tf.ones_like(discriminator_real_outputs),
192 tf.ones_like(discriminator_gen_outputs), discriminator_gen_outputs)
/external/tensorflow/tensorflow/contrib/gan/python/losses/python/
Dlosses_impl.py452 array_ops.ones_like(discriminator_real_outputs),
510 array_ops.ones_like(discriminator_gen_outputs),
605 array_ops.ones_like(discriminator_gen_outputs),

123456