Home
last modified time | relevance | path

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

12345

/external/tensorflow/tensorflow/python/ops/distributions/
Dmultinomial.py247 n_draws = array_ops.ones_like(
249 logits = array_ops.ones_like(
291 p = self.probs * array_ops.ones_like(
301 p = self.probs * array_ops.ones_like(
Dnormal.py219 scale = self.scale * array_ops.ones_like(self.loc)
223 return self.loc * array_ops.ones_like(self.scale)
229 return self.scale * array_ops.ones_like(self.loc)
Dbernoulli.py141 return (array_ops.ones_like(event) * logits,
142 array_ops.ones_like(logits) * event)
Dbeta.py244 expanded_concentration1 = array_ops.ones_like(
246 expanded_concentration0 = array_ops.ones_like(
/external/tensorflow/tensorflow/python/eager/
Dtape_test.py43 math_ops.matmul(array_ops.ones_like(b * dr), b, transpose_b=True),
45 math_ops.matmul(a, array_ops.ones_like(a) * dr, transpose_b=True)
96 array_ops.ones_like(aa),
110 array_ops.ones_like(aa),
124 array_ops.ones_like(aa),
/external/tensorflow/tensorflow/python/ops/
Dweights_broadcast_ops.py35 (values_shape_2d, array_ops.ones_like(values_shape_2d)), axis=1)
172 weights, array_ops.ones_like(values), name=scope)
175 weights, array_ops.ones_like(values), name=scope)
Ddefault_gradient.py56 def ones_like(t): function
61 return array_ops.ones_like(t)
Dconfusion_matrix.py189 values = (array_ops.ones_like(predictions, dtype)
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_v1.py330 array_ops.ones_like(inp.values, dtype=dtypes.float32),
383 array_ops.ones_like(inp.values, dtype=dtypes.float32),
388 weight = array_ops.ones_like(inp, dtype=dtypes.float32)
406 weight = array_ops.ones_like(inp, dtype=dtypes.float32)
/external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/
Dhvp_test.py57 helpers = tf.nest.map_structure(tf.ones_like, grads)
91 vector = [tf.ones_like(v) for v in model.trainable_variables]
123 vector = [tf.ones_like(v) for v in model.trainable_variables]
/external/tensorflow/tensorflow/core/transforms/constant_folding/tests/
Dmaterialize_constant_value_node.mlir7 …// CHECK-DAG: Const [%[[CTRL]]] name("ones_like") {{.*}} value = dense<1.000000e+00> {{.*}} -> (te…
8 …%OnesLike, %ctl_0 = OnesLike(%Placeholder) name("ones_like") {T = f32} : (tensor<1x2x3x4xf32>) -> …
Dneutral_element.mlir29 // CHECK: Const [%[[C]]] name("ones_like")
30 …%OnesLike, %ctl_23 = OnesLike(%Placeholder) name("ones_like") {T = f32} : (tensor<2x2xf32>) -> (te…
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_batch_gather_with_default_op.py152 array_ops.ones_like(params_shape.partitioned_dim_sizes[-1]),)
175 array_ops.ones_like(gather_dimension),) * (
Dragged_math_ops.py332 array_ops.ones_like(data.flat_values),
352 array_ops.ones_like(data.flat_values),
680 array_ops.ones_like(input_tensor.flat_values),
684 ones = array_ops.ones_like(input_tensor)
Dragged_tensor_test_ops.py86 array_ops.ones_like,
/external/tensorflow/tensorflow/python/tpu/tests/
Dtpu_embedding_base_test.py211 array_ops.ones_like(sparse.values, dtype=dtypes.float32) * weight)
251 array_ops.ones_like(sparse.values, dtype=dtypes.float32) * weight)
282 array_ops.ones_like(ragged.values, dtype=dtypes.float32) * weight)
317 array_ops.ones_like(t, dtype=dtypes.float32) * weight
346 array_ops.ones_like(t, dtype=dtypes.float32) * weight
Dtpu_embedding_v2_correctness_sequence_feature_test.py71 mid_level.apply_gradients(nest.map_structure(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/kernel_tests/distributions/
Dspecial_math_test.py85 all_true = np.ones_like(is_finite, dtype=np.bool_)
266 self.assertAllEqual(np.ones_like(v, dtype=np.bool_), v)
398 self.assertAllEqual(np.ones_like(x, dtype=np.bool_), x)
Ddirichlet_test.py91 np.ones_like(log_prob, dtype=np.bool_), np.isfinite(log_prob))
97 np.ones_like(log_prob, dtype=np.bool_), np.isfinite(log_prob))
/external/tensorflow/tensorflow/python/ops/losses/
Dlosses_impl.py119 array_ops.ones_like(weights))
191 loss, math_ops.reduce_sum(array_ops.ones_like(losses) * weights))
369 all_ones = array_ops.ones_like(labels)
/external/tensorflow/tensorflow/python/feature_column/
Dutils.py32 column_ids += array_ops.ones_like(column_ids)
/external/tensorflow/tensorflow/python/profiler/internal/
Dmodel_analyzer_testlib.py65 target = array_ops.ones_like(out)
/external/tensorflow/tensorflow/python/ops/structured/
Dstructured_array_ops.py276 @dispatch.dispatch_for_types(array_ops.ones_like, StructuredTensor)
277 def ones_like(tensor, dtype=None, name=None, optimize=True): function
/external/tensorflow/tensorflow/python/ops/linalg/
Dslicing.py125 param, param_ndims_to_matrix_ndims, array_ops.ones_like(batch_shape))

12345