Home
last modified time | relevance | path

Searched refs:add_n (Results 1 – 25 of 82) sorted by relevance

1234

/external/tensorflow/tensorflow/python/grappler/
Dcost_analyzer_test.py46 c = math_ops.add_n([a, b], name="c")
47 d = math_ops.add_n([b, c], name="d")
71 c = math_ops.add_n([a, b], name="c")
72 d = math_ops.add_n([b, c], name="d")
141 c = math_ops.add_n([a, b], name="c")
142 d = math_ops.add_n([b, c], name="d")
Dmemory_optimizer_test.py46 c = math_ops.add_n([a, b], name='c')
47 d = math_ops.add_n([b, c], name='d')
71 c = math_ops.add_n([a, b], name='c')
72 d = math_ops.add_n([b, c], name='d')
Dgraph_placer_test.py104 c = math_ops.add_n([a, b], name='c')
105 d = math_ops.add_n([b, c], name='d')
Dtf_optimizer_test.py43 c = math_ops.add_n([a, b], name='c')
44 d = math_ops.add_n([b, c], name='d')
Dmodel_analyzer_test.py38 d = math_ops.add_n([a, c], name="d")
/external/tensorflow/tensorflow/lite/kernels/
Dadd_n.cc24 namespace add_n { namespace
82 add_n::Prepare, add_n::Eval}; in Register_ADD_N()
/external/tensorflow/tensorflow/compiler/tests/
Dadd_n_test.py41 l = math_ops.add_n([l1, l2])
52 l = math_ops.add_n([l1, l2])
72 l = math_ops.add_n([l1, l2])
Dnary_ops_test.py57 self._testNAry(math_ops.add_n,
61 self._testNAry(math_ops.add_n,
65 self._testNAry(math_ops.add_n,
74 math_ops.add_n, [np.array([[1 + 2j, 2 - 3j, 3 + 4j]], dtype=dtype)],
78 math_ops.add_n, [
84 math_ops.add_n, [
/external/tensorflow/tensorflow/python/kernel_tests/
Daggregate_ops_test.py65 actual = self.evaluate(math_ops.add_n(data))
79 actual = sess.run(math_ops.add_n([data_ph] * count), {data_ph: data})
107 result = math_ops.add_n((variant_const_3, variant_const_3,
/external/tensorflow/tensorflow/python/ops/losses/
Dutil.py83 return math_ops.add_n(losses, name=name)
112 return math_ops.add_n(losses, name=name)
/external/tensorflow/tensorflow/python/ops/
Daccumulate_n_benchmark.py99 add_n = math_ops.add_n(inputs)
104 test_ops = (("AddN", add_n.op),
Dmath_ops_test.py357 partials.append(math_ops.add_n([constant_op.constant(1)]))
359 math_ops.add_n([constant_op.constant(1),
362 res = math_ops.add_n(partials) + constant_op.constant(0)
373 self.assertAllClose(sum(x), math_ops.add_n(tf_x).eval())
375 math_ops.add_n([tf_x[0]] * num_inputs).eval())
387 self.assertAllEqual(sum(x), math_ops.add_n(tf_x).eval())
389 math_ops.add_n([tf_x[0]] * num_inputs).eval())
400 addn = math_ops.add_n(input_vars)
415 self.assertAllEqual(slc_as_dense, math_ops.add_n([slc]).eval())
416 self.assertAllEqual(2 * slc_as_dense, math_ops.add_n([slc, slc]).eval())
/external/tensorflow/tensorflow/contrib/eager/python/examples/densenet/
Ddensenet_test.py123 keras_l2 = tf.add_n(conv.losses)
129 loss_with_keras_l2 = loss + tf.add_n(conv.losses)
136 keras_l2_after_update = tf.add_n(conv.losses)
145 regularization = tf.add_n(model.losses)
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dregularizers.py160 return math_ops.add_n(regularizer_tensors, name=name)
202 summed_penalty = math_ops.add_n(penalties, name=scope)
/external/tensorflow/tensorflow/python/distribute/
Dcross_device_utils.py155 grad = math_ops.add_n(grads)
403 scaled_grads, aux_devices, math_ops.add, math_ops.add_n)
407 scaled_grads, aux_devices, second_gather_devices, math_ops.add_n)
410 scaled_grads, aux_devices, math_ops.add_n)
645 def aggregate_tensors_or_indexed_slices(values, accumulation_fn=math_ops.add_n):
Dall_reduce_test.py159 simple_sum = math_ops.add_n(input_tensors)
162 sum_reduced = math_ops.add_n(output_tensors)
195 x, gather_devices, math_ops.add_n, un_op)
/external/tensorflow/tensorflow/contrib/slim/python/slim/data/
Dparallel_reader.py178 return math_ops.add_n(num_records, name=name)
190 return math_ops.add_n(num_work_units, name=name)
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dgmm_ops.py78 num_data = math_ops.add_n([array_ops.shape(inp)[0] for inp in data])
401 math_ops.add_n(self._points_in_k), axis=[0])
419 math_ops.add_n(self._w_mul_x), points_in_k_expanded + MEPS))
426 b = math_ops.add_n(self._w_mul_x2) / (points_in_k_expanded + MEPS)
Dclustering_ops.py529 new_clusters_centers = math_ops.add_n(cluster_sums) / (
530 math_ops.cast(math_ops.add_n(cluster_counts), cluster_sums[0].dtype) +
596 self._num_data = math_ops.add_n(
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_AccumulateNV2.pbtxt17 `tf.accumulate_n_v2` performs the same operation as `tf.add_n`, but does not
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_test.py113 math_ops.add_n([a_2_by_2, b_2_by_2]),
132 z2 = math_ops.add_n([a_2_by_2, b_2_by_2])
/external/tensorflow/tensorflow/python/framework/
Dfunction_def_to_graph_test.py45 sum_squares = math_ops.add_n(
47 sum_cubes = math_ops.add_n(
/external/tensorflow/tensorflow/contrib/integrate/python/ops/
Dodes.py74 return math_ops.add_n(
83 return math_ops.add_n([x * y for x, y in zip(xs, ys)], name=scope)
641 return math_ops.add_n([k1, 2 * k2, 2 * k3, k4]) * (dt_cast / 6)
/external/tensorflow/tensorflow/python/data/kernel_tests/
Doptional_test.py103 add_tensor = math_ops.add_n([opt1._variant_tensor,
113 add_tensor = math_ops.add_n([opt_none1._variant_tensor,
130 add_tensor = math_ops.add_n([opt3._variant_tensor,
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_dispatch_test.py402 {'op': math_ops.add_n,
412 def testListValuedElementwiseOp(self, inputs, op=math_ops.add_n,
486 self.evaluate(math_ops.add_n([x, y]))

1234