Home
last modified time | relevance | path

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

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorArgMax.h134 template<typename ReduceOp, typename Dims, typename XprType>
135 struct traits<TensorTupleReducerOp<ReduceOp, Dims, XprType> > : public traits<XprType>
147 template<typename ReduceOp, typename Dims, typename XprType>
148 struct eval<TensorTupleReducerOp<ReduceOp, Dims, XprType>, Eigen::Dense>
150 typedef const TensorTupleReducerOp<ReduceOp, Dims, XprType>& type;
153 template<typename ReduceOp, typename Dims, typename XprType>
154 struct nested<TensorTupleReducerOp<ReduceOp, Dims, XprType>, 1,
155 typename eval<TensorTupleReducerOp<ReduceOp, Dims, XprType> >::type>
157 typedef TensorTupleReducerOp<ReduceOp, Dims, XprType> type;
162 template<typename ReduceOp, typename Dims, typename XprType>
[all …]
DTensorForwardDeclarations.h38 template<typename ReduceOp, typename Dims, typename XprType> class TensorTupleReducerOp;
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.-reduce-op.pbtxt1 path: "tensorflow.distribute.ReduceOp"
3 is_instance: "<enum \'ReduceOp\'>"
6 mtype: "<enum \'ReduceOp\'>"
10 mtype: "<enum \'ReduceOp\'>"
Dtensorflow.distribute.pbtxt32 name: "ReduceOp"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.-reduce-op.pbtxt1 path: "tensorflow.distribute.ReduceOp"
3 is_instance: "<enum \'ReduceOp\'>"
6 mtype: "<enum \'ReduceOp\'>"
10 mtype: "<enum \'ReduceOp\'>"
Dtensorflow.distribute.pbtxt32 name: "ReduceOp"
/external/tensorflow/tensorflow/contrib/distribute/python/
Dcross_device_ops_test.py153 reduce_util.ReduceOp.MEAN,
159 reduce_util.ReduceOp.MEAN,
165 reduce_util.ReduceOp.SUM, per_replica,
170 reduce_util.ReduceOp.SUM,
179 reduce_util.ReduceOp.MEAN,
187 reduce_util.ReduceOp.SUM,
298 per_replica, devices[0], math_ops.add_n, reduce_util.ReduceOp.SUM)
319 reduce_op=[reduce_util.ReduceOp.SUM, reduce_util.ReduceOp.MEAN],
343 if reduce_op == reduce_util.ReduceOp.SUM:
347 assert reduce_op == reduce_util.ReduceOp.MEAN
[all …]
Dstrategy_test_lib.py146 reduce_util.ReduceOp.SUM, g, destinations=v)
198 reduce_util.ReduceOp.SUM, g, destinations=v)
567 return ctx.all_reduce(reduce_util.ReduceOp.SUM, value)
572 return ctx.all_reduce(reduce_util.ReduceOp.MEAN, value)
Dminimize_loss_test.py397 reduce_op=reduce_util.ReduceOp.MEAN)
407 reduce_op=reduce_util.ReduceOp.MEAN)
475 loss_tensor = distribution.reduce(reduce_util.ReduceOp.MEAN, loss_output)
Dcollective_all_reduce_strategy_test.py194 reduce_util.ReduceOp.SUM, g, destinations=v)
296 reduced_x = distribution.reduce(reduce_util.ReduceOp.MEAN, x)
Dparameter_server_strategy_test.py532 reduce_util.ReduceOp.SUM, g, destinations=v)
Dmirrored_strategy_multigpu_test.py105 reduced = distribution.reduce(reduce_util.ReduceOp.SUM, result)
/external/tensorflow/tensorflow/python/distribute/
Dreduce_util.py28 class ReduceOp(enum.Enum): class
45 variable_scope.VariableAggregation.SUM: ReduceOp.SUM,
46 variable_scope.VariableAggregation.MEAN: ReduceOp.MEAN,
Ddistribute_lib.py89 return reduce_util.ReduceOp.SUM
1157 assert (reduce_op == reduce_util.ReduceOp.SUM or
1158 reduce_op == reduce_util.ReduceOp.MEAN)
1505 if reduce_op in [reduce_util.ReduceOp.SUM, reduce_util.ReduceOp.MEAN]:
Dcross_device_ops.py85 if reduce_op == reduce_util.ReduceOp.MEAN:
216 if reduce_op == reduce_util.ReduceOp.MEAN:
219 elif reduce_op != reduce_util.ReduceOp.SUM:
457 if reduce_op == reduce_util.ReduceOp.MEAN:
Dtpu_strategy.py443 if reduce_op == reduce_util.ReduceOp.MEAN:
446 elif reduce_op != reduce_util.ReduceOp.SUM:
466 if reduce_op == reduce_util.ReduceOp.MEAN:
Dcollective_all_reduce_strategy.py432 reduce_op == reduce_util.ReduceOp.MEAN):
Dvalues.py628 reduce_op = reduce_util.ReduceOp.from_variable_aggregation(aggregation)
1267 reduce_util.ReduceOp.from_variable_aggregation(self.aggregation), self)
Dmirrored_strategy.py696 reduce_op == reduce_util.ReduceOp.MEAN):
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_distributed.py252 reduce_op = ds_reduce_util.ReduceOp.SUM
256 reduce_op = ds_reduce_util.ReduceOp.MEAN
Ddistributed_training_utils.py108 loss = distribution_strategy.reduce(reduce_util.ReduceOp.SUM,
/external/tensorflow/tensorflow/python/training/
Dmoving_averages.py100 ds_reduce_util.ReduceOp.MEAN, value, v)
Doptimizer.py645 ds_reduce_util.ReduceOp.SUM, grads_and_vars)
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Doptimizer_v2.py416 ds_reduce_util.ReduceOp.SUM, grads_and_vars)
/external/tensorflow/tensorflow/contrib/optimizer_v2/
Doptimizer_v2.py915 ds_reduce_util.ReduceOp.SUM, grads_and_vars)