Searched refs:CrossReplicaSum (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | CrossReplicaSum.pbtxt | 2 name: "CrossReplicaSum"
27 name: "CrossReplicaSum"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_CrossReplicaSum.pbtxt | 2 graph_op_name: "CrossReplicaSum"
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 1056 def CrossReplicaSum(self, operand, replica_groups=None): member in ComputationBuilder
1070 return ops.CrossReplicaSum(operand, replica_groups_protos)
|
| D | xla.cc | 828 &CrossReplicaSum)); in PYBIND11_MODULE()
|
| D | xla_client_test.py | 706 c.CrossReplicaSum(c.Constant(lhs))
723 c.CrossReplicaSum(c.Constant(lhs), [[0]])
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 511 XlaOp CrossReplicaSum(XlaOp operand,
922 friend XlaOp CrossReplicaSum(XlaOp operand,
1653 XlaOp CrossReplicaSum(XlaOp operand,
|
| D | xla_builder.cc | 2083 XlaOp XlaBuilder::CrossReplicaSum( in CrossReplicaSum() function in xla::XlaBuilder
3301 XlaOp CrossReplicaSum(const XlaOp operand, in CrossReplicaSum() function
3303 return operand.builder()->CrossReplicaSum(operand, replica_groups); in CrossReplicaSum()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_patterns.td | 273 // CrossReplicaSum op patterns.
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 848 ## CrossReplicaSum section in Operation Semantics
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf.mlir | 2705 …%result = "tf.CrossReplicaSum" (%input, %replica_groups) : (tensor<10xf32>, tensor<2x4xi32>) -> te…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 860 name: "CrossReplicaSum"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 860 name: "CrossReplicaSum"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 1086 def TF_CrossReplicaSumOp : TF_Op<"CrossReplicaSum", [AllTypesMatch<["input", "output"]>, NoSideEffe…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 44984 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"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 8733 name: "CrossReplicaSum"
|