Home
last modified time | relevance | path

Searched refs:reduce_op (Results 1 – 25 of 44) sorted by relevance

12

/external/tensorflow/tensorflow/python/distribute/
Dcross_device_ops.py72 def reduce_non_distributed_value(reduce_op, device_map, value, destinations): argument
85 if reduce_op == reduce_util.ReduceOp.MEAN:
94 "the given reduce op %s." % (value, reduce_op))
205 reduce_op): argument
216 if reduce_op == reduce_util.ReduceOp.MEAN:
219 elif reduce_op != reduce_util.ReduceOp.SUM:
231 def reduce(self, reduce_op, per_replica_value, destinations): argument
254 return self.reduce_implementation(reduce_op, per_replica_value,
257 def batch_reduce(self, reduce_op, value_destination_pairs): argument
287 return self.batch_reduce_implementation(reduce_op, value_destination_pairs)
[all …]
Ddistribute_lib.py501 def reduce(self, reduce_op, value): argument
513 return self._extended._reduce(reduce_op, value) # pylint: disable=protected-access
1133 def _reduce(self, reduce_op, value): argument
1136 self._reduce_to(reduce_op, value,
1139 def reduce_to(self, reduce_op, value, destinations): argument
1156 assert not isinstance(reduce_op, variable_scope.VariableAggregation)
1157 assert (reduce_op == reduce_util.ReduceOp.SUM or
1158 reduce_op == reduce_util.ReduceOp.MEAN)
1159 return self._reduce_to(reduce_op, value, destinations)
1161 def _reduce_to(self, reduce_op, value, destinations): argument
[all …]
Dreduce_util.py49 reduce_op = mapping.get(aggregation)
50 if not reduce_op:
53 return reduce_op
Dinput_lib.py599 def set_last_step_output(self, name, output, reduce_op=None): argument
620 self._last_step_outputs_reduce_ops[name] = reduce_op
621 if reduce_op is None:
625 self._last_step_outputs[name] = distribution.reduce(reduce_op, output)
627 assert reduce_op is not None
629 self._last_step_outputs[name] = distribution.reduce(reduce_op, value)
633 self._last_step_outputs_reduce_ops[name] = reduce_op
Dtpu_strategy.py441 def _reduce_to(self, reduce_op, value, destinations): argument
443 if reduce_op == reduce_util.ReduceOp.MEAN:
446 elif reduce_op != reduce_util.ReduceOp.SUM:
457 reduce_op, self._device_map, value, destinations)
466 if reduce_op == reduce_util.ReduceOp.MEAN:
662 …for name, reduce_op in ctx._last_step_outputs_reduce_ops.items(): # pylint: disable=protected-acc…
669 if reduce_op is not None:
Dparameter_server_strategy.py380 def _reduce_to(self, reduce_op, value, destinations): argument
385 reduce_op, self._device_map, value, destinations)
387 reduce_op, value, destinations=destinations)
389 def _batch_reduce_to(self, reduce_op, value_destination_pairs): argument
392 return self._cross_device_ops.batch_reduce(reduce_op,
Dmirrored_strategy.py634 …for name, reduce_op in ctx._last_step_outputs_reduce_ops.items(): # pylint: disable=protected-acc…
638 if reduce_op is None:
694 def _reduce_to(self, reduce_op, value, destinations): argument
696 reduce_op == reduce_util.ReduceOp.MEAN):
705 reduce_op, self._device_map, value, destinations)
707 reduce_op, value, destinations=destinations)
709 def _batch_reduce_to(self, reduce_op, value_destination_pairs): argument
711 reduce_op, value_destination_pairs)
Dcollective_all_reduce_strategy.py430 def _reduce_to(self, reduce_op, value, destinations): argument
432 reduce_op == reduce_util.ReduceOp.MEAN):
449 reduce_op, self._device_map, value, destinations)
451 reduce_op, value, destinations=destinations)
Done_device_strategy.py146 def _reduce_to(self, reduce_op, value, destinations): argument
147 del reduce_op, destinations
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.-cross-device-ops.pbtxt11 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
15 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
27 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
31 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
Dtensorflow.distribute.-nccl-all-reduce.pbtxt13 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
17 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
29 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
33 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
Dtensorflow.distribute.-reduction-to-one-device.pbtxt12 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
16 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
28 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
32 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
Dtensorflow.distribute.-hierarchical-copy-all-reduce.pbtxt13 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
17 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
29 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
33 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
Dtensorflow.distribute.-strategy-extended.pbtxt39 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
71 …argspec: "args=[\'self\', \'reduce_op\', \'value\', \'destinations\'], varargs=None, keywords=None…
Dtensorflow.distribute.-replica-context.pbtxt27 argspec: "args=[\'self\', \'reduce_op\', \'value\'], varargs=None, keywords=None, defaults=None"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.-cross-device-ops.pbtxt11 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
15 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
27 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
31 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
Dtensorflow.distribute.-nccl-all-reduce.pbtxt13 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
17 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
29 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
33 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
Dtensorflow.distribute.-reduction-to-one-device.pbtxt12 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
16 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
28 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
32 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
Dtensorflow.distribute.-hierarchical-copy-all-reduce.pbtxt13 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
17 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
29 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
33 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\'], varargs=None, k…
Dtensorflow.distribute.-strategy-extended.pbtxt39 …argspec: "args=[\'self\', \'reduce_op\', \'value_destination_pairs\'], varargs=None, keywords=None…
71 …argspec: "args=[\'self\', \'reduce_op\', \'value\', \'destinations\'], varargs=None, keywords=None…
Dtensorflow.distribute.-replica-context.pbtxt27 argspec: "args=[\'self\', \'reduce_op\', \'value\'], varargs=None, keywords=None, defaults=None"
/external/tensorflow/tensorflow/contrib/distribute/python/
Dcross_device_ops_test.py319 reduce_op=[reduce_util.ReduceOp.SUM, reduce_util.ReduceOp.MEAN],
323 def testIndexedSlicesAllReduce(self, cross_device_ops_instance, reduce_op, argument
335 reduce_op, [(per_replica, per_replica)])
338 reduce_op, per_replica, per_replica)
343 if reduce_op == reduce_util.ReduceOp.SUM:
347 assert reduce_op == reduce_util.ReduceOp.MEAN
548 def _reduce(test_object, reduce_op, per_replica, destinations): argument
555 return test_object.extended.reduce_to(reduce_op, per_replica,
558 return test_object.reduce(reduce_op, per_replica, destinations)
560 def _batch_reduce(test_object, reduce_op, value_destination_pairs): argument
[all …]
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_math_ops.py399 def _ragged_reduce_aggregate(reduce_op, argument
441 return reduce_op(rt_input, axis, name=name)
455 return reduce_op(rt_input.flat_values, None, name=name)
469 inner_reduced = _ragged_reduce_aggregate(reduce_op, unsorted_segment_op,
471 return _ragged_reduce_aggregate(reduce_op, unsorted_segment_op,
497 _ragged_reduce_aggregate(reduce_op, unsorted_segment_op,
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorArgMax.h174 const ReduceOp& reduce_op,
177 : m_xpr(expr), m_reduce_op(reduce_op), m_return_dim(return_dim), m_reduce_dims(reduce_dims) {}
184 const ReduceOp& reduce_op() const { return m_reduce_op; }
224 m_impl(op.expression().index_tuples().reduce(op.reduce_dims(), op.reduce_op()), device),
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_distributed.py252 reduce_op = ds_reduce_util.ReduceOp.SUM
256 reduce_op = ds_reduce_util.ReduceOp.MEAN
257 ctx.set_last_step_output(label, output, reduce_op)

12