| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | CrossReplicaSum.pbtxt | 2 name: "CrossReplicaSum"
27 name: "CrossReplicaSum"
54 name: "CrossReplicaSum"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | CrossReplicaSum.pbtxt | 2 name: "CrossReplicaSum"
27 name: "CrossReplicaSum"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | tpu_space_to_depth_pass.mlir | 78 …%8 = "tf.CrossReplicaSum"(%7, %1) : (tensor<7x7x3x64xf32>, tensor<1x1xi32>) -> tensor<7x7x3x64xf32>
91 …try_function = {control_outputs = "IteratorGetNext,IteratorGetNext_1,CrossReplicaSum,AssignAddVari…
145 %23 = "tf.CrossReplicaSum"(%22, %3) : (tensor<f32>, tensor<1x2xi32>) -> tensor<f32>
152 %30 = "tf.CrossReplicaSum"(%29, %3) : (tensor<f32>, tensor<1x2xi32>) -> tensor<f32>
154 %32 = "tf.CrossReplicaSum"(%1, %3) : (tensor<f32>, tensor<1x2xi32>) -> tensor<f32>
157 %35 = "tf.CrossReplicaSum"(%0, %3) : (tensor<f32>, tensor<1x2xi32>) -> tensor<f32>
168 …try_function = {control_outputs = "IteratorGetNext,IteratorGetNext_1,CrossReplicaSum,AssignAddVari…
219 %23 = "tf.CrossReplicaSum"(%22, %3) : (tensor<f32>, tensor<1x2xi32>) -> tensor<f32>
226 %30 = "tf.CrossReplicaSum"(%29, %3) : (tensor<f32>, tensor<1x2xi32>) -> tensor<f32>
228 %32 = "tf.CrossReplicaSum"(%1, %3) : (tensor<f32>, tensor<1x2xi32>) -> tensor<f32>
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_CrossReplicaSum.pbtxt | 2 graph_op_name: "CrossReplicaSum"
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | cross_replica_ops.cc | 61 ctx->SetOutput(0, xla::CrossReplicaSum(ctx->Input(0), replica_groups)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 84 &CrossReplicaSum), in BuildOpsSubmodule()
|
| D | xla_client_test.py | 661 ops.CrossReplicaSum(ops.Constant(c, lhs))
678 ops.CrossReplicaSum(
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 730 XlaOp CrossReplicaSum(XlaOp operand,
1295 friend XlaOp CrossReplicaSum(XlaOp operand,
2185 XlaOp CrossReplicaSum(XlaOp operand,
|
| D | xla_builder.cc | 2751 XlaOp XlaBuilder::CrossReplicaSum( in CrossReplicaSum() function in xla::XlaBuilder
4664 XlaOp CrossReplicaSum(const XlaOp operand, in CrossReplicaSum() function
4666 return operand.builder()->CrossReplicaSum(operand, replica_groups); in CrossReplicaSum()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_patterns.td | 286 // CrossReplicaSum op patterns.
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 1559 ### tf.CrossReplicaSum
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 899 ## CrossReplicaSum section in Operation Semantics
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 940 name: "CrossReplicaSum"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 940 name: "CrossReplicaSum"
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf.mlir | 3872 …%result = "tf.CrossReplicaSum" (%input, %replica_groups) : (tensor<10xf32>, tensor<2x4xi32>) -> te…
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 16575 name: "CrossReplicaSum"
16600 name: "CrossReplicaSum"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 45556 func CrossReplicaSum(scope *Scope, input tf.Output, group_assignment tf.Output) (output tf.Output) { func
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 2487 def TF_CrossReplicaSumOp : TF_Op<"CrossReplicaSum", [NoSideEffect, TF_AllTypesMatch<["input", "outp…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 9489 name: "CrossReplicaSum"
|