Home
last modified time | relevance | path

Searched refs:add_n (Results 1 – 25 of 80) 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')
72 c = math_ops.add_n([a, b], name='c')
73 d = math_ops.add_n([b, c], name='d')
Dtf_optimizer_test.py44 c = math_ops.add_n([a, b], name='c')
45 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/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/lite/kernels/
Dadd_n.cc26 namespace add_n { namespace
96 add_n::Prepare, add_n::Eval}; in Register_ADD_N()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dadd_n.cc26 namespace add_n { namespace
95 add_n::Prepare, add_n::Eval}; in Register_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})
110 variant_sums = [math_ops.add_n([variant_consts[i] for i in x])
/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.py373 partials.append(math_ops.add_n([constant_op.constant(1)]))
375 math_ops.add_n([constant_op.constant(1),
378 res = math_ops.add_n(partials) + constant_op.constant(0)
388 self.assertAllClose(sum(x), math_ops.add_n(tf_x))
390 math_ops.add_n([tf_x[0]] * num_inputs))
401 self.assertAllEqual(sum(x), math_ops.add_n(tf_x))
403 math_ops.add_n([tf_x[0]] * num_inputs))
417 addn = math_ops.add_n(input_vars)
420 addn = math_ops.add_n(input_vars)
434 self.assertAllEqual(slc_as_dense, math_ops.add_n([slc]))
[all …]
Dclustering_ops.py530 new_clusters_centers = math_ops.add_n(cluster_sums) / (
531 math_ops.cast(math_ops.add_n(cluster_counts), cluster_sums[0].dtype) +
597 self._num_data = math_ops.add_n(
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dload-quantization-recipe.mlir25 // CHECK-NEXT: %[[in4:.*]] = "tfl.add_n"(%[[in1]], %[[in2]], %[[in3]])
43 // CHECK-NEXT: %[[fo4:.*]] = "tfl.add_n"(%[[fo1]], %[[fo2]], %[[fo3]])
59 // CHECK-NEXT: %[[ce3:.*]] = "tfl.add_n"(%[[ce1]], %[[ce2]])
84 // CHECK-NEXT: %[[ou4:.*]] = "tfl.add_n"(%[[ou1]], %[[ou2]], %[[ou3]])
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_AddN.pbtxt9 tf.math.add_n(x) ==> 26
Dapi_def_AccumulateNV2.pbtxt17 `tf.accumulate_n_v2` performs the same operation as `tf.add_n`, but does not
/external/tensorflow/tensorflow/python/ops/losses/
Dutil.py232 return math_ops.add_n(losses, name=name)
266 return math_ops.add_n(losses, name=name)
/external/tensorflow/tensorflow/python/distribute/v1/
Dall_reduce_test.py158 simple_sum = math_ops.add_n(input_tensors)
161 sum_reduced = math_ops.add_n(output_tensors)
194 x, gather_devices, math_ops.add_n, un_op)
/external/tensorflow/tensorflow/lite/testing/op_tests/
Dadd_n.py66 out = tf.add_n(input_tensors)
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_dispatch_test.py312 {'op': math_ops.add_n,
322 def testListValuedElementwiseOp(self, inputs, op=math_ops.add_n,
392 self.evaluate(math_ops.add_n([x, y]))
707 math_ops.add_n([rt, rt]).uniform_row_length)
/external/tensorflow/tensorflow/python/framework/
Dfunction_def_to_graph_test.py46 sum_squares = math_ops.add_n(
48 sum_cubes = math_ops.add_n(
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dreshape_transpose_test.py60 return math_ops.add_n(outputs, name="output_0")
/external/tensorflow/tensorflow/python/eager/
Dfunction_argument_naming_test.py203 return x + math_ops.add_n(list(args) + list(kwargs.values()))
231 return x + math_ops.add_n(list(args))
/external/tensorflow/tensorflow/python/data/kernel_tests/
Doptional_test.py187 add_tensor = math_ops.add_n(
195 add_tensor = math_ops.add_n(
212 add_tensor = math_ops.add_n(
/external/tensorflow/tensorflow/python/keras/engine/
Dcompile_utils.py235 reg_loss = math_ops.add_n(regularization_losses)
242 total_loss_metric_value = math_ops.add_n(loss_metric_values)
247 total_loss = math_ops.add_n(loss_values)
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale_benchmark.py102 return math_ops.add_n(var_list)

1234