| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_CrossReplicaSum.pbtxt | 2 graph_op_name: "CrossReplicaSum"
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 466 XlaOp CrossReplicaSum(const XlaOp& operand,
469 XlaOp CrossReplicaSum(
884 friend XlaOp CrossReplicaSum(const XlaOp& operand,
886 friend XlaOp CrossReplicaSum(const XlaOp& operand,
1567 XlaOp CrossReplicaSum(const XlaOp& operand,
1587 XlaOp CrossReplicaSum(
|
| D | xla_builder.cc | 2120 XlaOp XlaBuilder::CrossReplicaSum( in CrossReplicaSum() function in xla::XlaBuilder
2129 return CrossReplicaSum(operand, computation, replica_groups, in CrossReplicaSum()
2134 XlaOp XlaBuilder::CrossReplicaSum( in CrossReplicaSum() function in xla::XlaBuilder
3232 XlaOp CrossReplicaSum(const XlaOp& operand, in CrossReplicaSum() function
3234 return operand.builder()->CrossReplicaSum(operand, replica_groups); in CrossReplicaSum()
3237 XlaOp CrossReplicaSum(const XlaOp& operand, const XlaComputation& computation, in CrossReplicaSum() function
3240 return operand.builder()->CrossReplicaSum(operand, computation, in CrossReplicaSum()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 510 LocalOp ComputationBuilder::CrossReplicaSum( in CrossReplicaSum() function in xla::swig::ComputationBuilder
512 return xla::CrossReplicaSum(operand.op(), replica_groups); in CrossReplicaSum()
|
| D | local_computation_builder.h | 259 LocalOp CrossReplicaSum(const LocalOp& operand,
|
| D | local_computation_builder.i | 306 %unignore xla::swig::ComputationBuilder::CrossReplicaSum;
|
| D | xla_client.py | 1241 def CrossReplicaSum(self, operand, replica_groups=None): member in ComputationBuilder
1260 return self._client.CrossReplicaSum(operand, replica_groups)
|
| D | xla_client_test.py | 606 c.CrossReplicaSum(c.Constant(lhs))
623 c.CrossReplicaSum(c.Constant(lhs), [[0]])
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 781 ## CrossReplicaSum section in Operation Semantics
784 [`XlaBuilder::CrossReplicaSum`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla_…
788 <b> `CrossReplicaSum(operand)` </b>
808 Computing the result of CrossReplicaSum requires having one input from each
809 replica, so if one replica executes a CrossReplicaSum node more times than
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 728 name: "CrossReplicaSum"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 728 name: "CrossReplicaSum"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 20923 func CrossReplicaSum(scope *Scope, input tf.Output, group_assignment tf.Output) (output tf.Output) { func
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 16575 name: "CrossReplicaSum"
16600 name: "CrossReplicaSum"
|
| D | ops_history.v1.pbtxt | 16633 name: "CrossReplicaSum"
16658 name: "CrossReplicaSum"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 7196 name: "CrossReplicaSum"
|