Searched refs:accumulate_n (Results 1 – 11 of 11) sorted by relevance
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | accumulate_n_test.py | 41 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 …]
|
D | accumulate_n_eager_test.py | 39 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])
|
D | cwise_ops_test.py | 1038 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/ |
D | api_def_AccumulateNV2.pbtxt | 22 Unlike the original `accumulate_n`, `accumulate_n_v2` is differentiable.
|
/external/tensorflow/tensorflow/python/ops/ |
D | math_ops_test.py | 337 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())
|
D | gradients_util.py | 1028 out_grads[i] = math_ops.accumulate_n(out_grad)
|
D | math_ops.py | 2803 def accumulate_n(inputs, shape=None, tensor_dtype=None, name=None): function
|
/external/tensorflow/tensorflow/contrib/distribute/python/ |
D | cross_device_ops_test.py | 217 accumulation_fn=math_ops.accumulate_n)),
|
/external/tensorflow/tensorflow/tools/api/golden/v2/ |
D | tensorflow.math.pbtxt | 8 name: "accumulate_n"
|
/external/tensorflow/tensorflow/tools/api/golden/v1/ |
D | tensorflow.math.pbtxt | 8 name: "accumulate_n"
|
D | tensorflow.pbtxt | 696 name: "accumulate_n"
|