Home
last modified time | relevance | path

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

12345

/external/tensorflow/tensorflow/python/keras/layers/
Dconvolutional_recurrent_test.py112 out1 = model.predict(np.ones_like(inputs))
115 model.train_on_batch(np.ones_like(inputs),
117 out2 = model.predict(np.ones_like(inputs))
125 out3 = model.predict(np.ones_like(inputs))
130 out4 = model.predict(np.ones_like(inputs))
134 out5 = model.predict(np.ones_like(inputs))
/external/tensorflow/tensorflow/python/eager/
Dtape_test.py46 math_ops.matmul(array_ops.ones_like(b * dr), b, transpose_b=True),
48 math_ops.matmul(a, array_ops.ones_like(a) * dr, transpose_b=True)
99 array_ops.ones_like(aa),
113 array_ops.ones_like(aa),
127 array_ops.ones_like(aa),
/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)
Dbeta.py248 expanded_concentration1 = array_ops.ones_like(
250 expanded_concentration0 = array_ops.ones_like(
Duniform.py186 array_ops.ones_like(broadcasted_x) / self.range()))
/external/tensorflow/tensorflow/python/ops/
Dweights_broadcast_ops.py39 (values_shape_2d, array_ops.ones_like(values_shape_2d)), axis=1)
175 weights, array_ops.ones_like(values), name=scope)
178 weights, array_ops.ones_like(values), name=scope)
Ddefault_gradient.py60 def ones_like(t): function
65 return array_ops.ones_like(t)
Dconfusion_matrix.py193 values = (array_ops.ones_like(predictions, dtype)
/external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/
Dhvp_test.py62 helpers = tf.nest.map_structure(tf.ones_like, grads)
96 vector = [tf.ones_like(v) for v in model.trainable_variables]
128 vector = [tf.ones_like(v) for v in model.trainable_variables]
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_batch_gather_with_default_op.py156 array_ops.ones_like(params_shape.partitioned_dim_sizes[-1]),)
179 array_ops.ones_like(gather_dimension),) * (
Dragged_math_ops.py310 array_ops.ones_like(data.flat_values),
326 array_ops.ones_like(data.flat_values),
618 array_ops.ones_like(input_tensor.flat_values),
622 ones = array_ops.ones_like(input_tensor)
Dragged_tensor_test_ops.py73 array_ops.ones_like,
/external/tensorflow/tensorflow/c/eager/
Dtape.h619 Gradient* ones_like = nullptr; in InitialGradients() local
621 op_it->second.output_tensor_info[j], &ones_like)); in InitialGradients()
622 (*result)[id].push_back(ones_like); in InitialGradients()
637 Gradient* ones_like = nullptr; in InitialGradients() local
639 vspace.BuildOnesLike(source_tensor->second, &ones_like)); in InitialGradients()
640 (*result)[id].push_back(ones_like); in InitialGradients()
/external/tensorflow/tensorflow/python/ops/losses/
Dlosses_impl.py122 array_ops.ones_like(weights))
195 loss, math_ops.reduce_sum(array_ops.ones_like(losses) * weights))
362 all_ones = array_ops.ones_like(labels)
/external/tensorflow/tensorflow/python/kernel_tests/distributions/
Dspecial_math_test.py89 all_true = np.ones_like(is_finite, dtype=np.bool)
270 self.assertAllEqual(np.ones_like(v, dtype=np.bool), v)
402 self.assertAllEqual(np.ones_like(x, dtype=np.bool), x)
Ddirichlet_test.py95 np.ones_like(log_prob, dtype=np.bool), np.isfinite(log_prob))
101 np.ones_like(log_prob, dtype=np.bool), np.isfinite(log_prob))
/external/tensorflow/tensorflow/python/feature_column/
Dutils.py36 column_ids += array_ops.ones_like(column_ids)
/external/tensorflow/tensorflow/python/keras/
Dactivations_test.py120 return np.log(np.ones_like(x) + np.exp(x))
131 return np.divide(x, np.ones_like(x) + np.absolute(x))
/external/tensorflow/tensorflow/python/profiler/internal/
Dmodel_analyzer_testlib.py69 target = array_ops.ones_like(out)
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_v2_correctness_test.py236 array_ops.ones_like(sparse.values, dtype=dtypes.float32) * weight)
278 weights = [array_ops.ones_like(t, dtype=dtypes.float32) * weight
509 mid_level.apply_gradients(nest.map_structure(array_ops.ones_like,
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/
Dhashing.py193 values = math_ops.add(values, array_ops.ones_like(values))
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_kronecker.py331 [array_ops.ones_like(self.batch_shape_tensor()), [1, 1]], 0)
447 [array_ops.ones_like(self.batch_shape_tensor()), [1, 1]], 0)
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/benchmarks/
Dnormalization_adapt_benchmark.py48 ones = array_ops.ones_like(values, dtype=dtypes.int32)

12345