Home
last modified time | relevance | path

Searched refs:accumulate_n (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Daccumulate_n_test.py41 self.assertAllClose(sum(x), math_ops.accumulate_n(tf_x).eval())
43 math_ops.accumulate_n([tf_x[0]] * 5).eval())
51 self.assertAllEqual(sum(x), math_ops.accumulate_n(tf_x).eval())
53 math_ops.accumulate_n([tf_x[0]] * 6).eval())
59 acc = math_ops.accumulate_n([x0, x0], shape=[None])
71 accum_n = math_ops.accumulate_n(input_vars)
91 tf_val = math_ops.accumulate_n(random_tensors)
100 tf_val = math_ops.accumulate_n([])
108 math_ops.accumulate_n([a, b], shape=[2, 2]) # Should be shape=[]
115 math_ops.accumulate_n([a, b])
[all …]
Daccumulate_n_eager_test.py39 answer = math_ops.accumulate_n([forty, two])
46 self.assertAllClose(sum(x), math_ops.accumulate_n(tf_x))
48 math_ops.accumulate_n([tf_x[0]] * 5))
60 return math_ops.accumulate_n([first, second, third])
Dcwise_ops_test.py1038 tf_val = math_ops.accumulate_n(random_tensors)
1047 tf_val = math_ops.accumulate_n([])
1055 math_ops.accumulate_n([a, b], shape=[2, 2]) # Should be shape=[]
1062 math_ops.accumulate_n([a, b], tensor_dtype=np.int32)
1069 math_ops.accumulate_n([a], tensor_dtype=np.int32)
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_AccumulateNV2.pbtxt22 Unlike the original `accumulate_n`, `accumulate_n_v2` is differentiable.
/external/tensorflow/tensorflow/python/ops/
Dmath_ops_test.py337 self.assertAllClose(sum(x), math_ops.accumulate_n(tf_x).eval())
338 self.assertAllClose(x[0] * 5, math_ops.accumulate_n([tf_x[0]] * 5).eval())
346 self.assertAllEqual(sum(x), math_ops.accumulate_n(tf_x).eval())
347 self.assertAllEqual(x[0] * 6, math_ops.accumulate_n([tf_x[0]] * 6).eval())
Dgradients_util.py1028 out_grads[i] = math_ops.accumulate_n(out_grad)
Dmath_ops.py2803 def accumulate_n(inputs, shape=None, tensor_dtype=None, name=None): function
/external/tensorflow/tensorflow/contrib/distribute/python/
Dcross_device_ops_test.py217 accumulation_fn=math_ops.accumulate_n)),
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.math.pbtxt8 name: "accumulate_n"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.math.pbtxt8 name: "accumulate_n"
Dtensorflow.pbtxt696 name: "accumulate_n"