Searched refs:TPUReplicatedInput (Results 1 – 18 of 18) sorted by relevance
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
D | tpu_cluster_formation.mlir | 181 // Test one replica cluster results in removing of TPUReplicatedInput and 186 %ri = "tf.TPUReplicatedInput"(%arg0) : (tensor<i1>) -> tensor<i1> 210 // CHECK-NOT: "tf.TPUReplicatedInput" 216 // TPUReplicatedInput and TPUReplicatedOutput nodes will be replaced by the 222 %ri_0 = "tf.TPUReplicatedInput"(%arg0, %0) : (tensor<i1>, tensor<i1>) -> tensor<i1> 224 %ri_1 = "tf.TPUReplicatedInput"(%1, %arg1) : (tensor<i32>, tensor<i32>) -> tensor<i32> 258 // TPUReplicatedInput and TPUReplicatedOutput nodes will be replaced by the 267 %2 = "tf.TPUReplicatedInput"(%arg0, %arg2) : (!rtype, !rtype) -> !rtype 268 %3 = "tf.TPUReplicatedInput"(%arg1, %arg3) : (!rtype, !rtype) -> !rtype 270 %5 = "tf.TPUReplicatedInput"(%0, %1) : (tensor<i32>, tensor<i32>) -> tensor<i32> [all …]
|
D | tpu_reorder_replicate_and_partitioned_inputs.mlir | 6 // CHECK: [[RI_0:%.*]] = "tf.TPUReplicatedInput"([[ARG0]], [[ARG2]]) 7 // CHECK: [[RI_1:%.*]] = "tf.TPUReplicatedInput"([[ARG1]], [[ARG3]]) 11 …%ri = "tf.TPUReplicatedInput"(%pi_0, %pi_1) : (tensor<!tf.resource<tensor<10x3xf32>>>, tensor<!tf.… 19 // CHECK: [[RI_0:%.*]] = "tf.TPUReplicatedInput"([[ARG0]], [[ARG2]]) 20 // CHECK: [[RI_1:%.*]] = "tf.TPUReplicatedInput"([[ARG1]], [[ARG3]]) 24 …%ri = "tf.TPUReplicatedInput"(%pi_0, %pi_1) : (tensor<!tf.resource<tensor<10x3xf32>>>, tensor<!tf.… 39 // CHECK: [[RI:%.*]] = "tf.TPUReplicatedInput"([[ARG0]], [[ARG1]]) 40 …%ri = "tf.TPUReplicatedInput"(%arg0, %arg1) : (tensor<!tf.resource<tensor<10x3xf32>>>, tensor<!tf.… 51 …%ri = "tf.TPUReplicatedInput"(%pi_0, %pi_1) : (tensor<!tf.resource<tensor<10x3xf32>>>, tensor<!tf.… 61 …%ri = "tf.TPUReplicatedInput"(%pi_0, %pi_1) : (tensor<!tf.resource<tensor<10x3xf32>>>, tensor<!tf.… [all …]
|
D | graph_pruning_preserve_ops.mlir | 1 …ps-to-preserve="tf.TPUReplicateMetadata,tf.TPUCompilationResult,tf.TPUReplicatedInput,tf.TPUReplic… 39 // CHECK: "tf.TPUReplicatedInput" 40 …%4, %5 = tf_executor.island(%1) wraps "tf.TPUReplicatedInput"(%2) {index = -1 : i64, is_mirrored_v…
|
D | shape_inference.mlir | 178 // CHECK: "tf.TPUReplicatedInput" 180 %ri = "tf.TPUReplicatedInput"(%2) : (tensor<*x!tf.resource>) -> tensor<*x!tf.resource>
|
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
D | TPUReplicatedInput.pbtxt | 2 name: "TPUReplicatedInput" 24 name: "TPUReplicatedInput" 53 name: "TPUReplicatedInput"
|
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
D | TPUReplicatedInput.pbtxt | 2 name: "TPUReplicatedInput" 24 name: "TPUReplicatedInput" 60 name: "TPUReplicatedInput"
|
/external/tensorflow/tensorflow/core/api_def/base_api/ |
D | api_def_TPUReplicatedInput.pbtxt | 2 graph_op_name: "TPUReplicatedInput" 13 %replicated_input = "tf.TPUReplicatedInput"(%a, %b)
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tpu_bridge_v1/ |
D | end_to_end.mlir | 14 …%outputs_5, %control_6 = tf_executor.island wraps "tf.TPUReplicatedInput"(%outputs_3) {N = 1 : i64… 18 …%outputs_13, %control_14 = tf_executor.island wraps "tf.TPUReplicatedInput"(%outputs_11) {N = 1 : …
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
D | tf_passes.td | 138 optionally `tf.TPUReplicatedInput` and `tf.TPUReplicatedOutput` ops feeding in 142 `tf.TPUReplicatedInput` and `tf.TPUReplicatedOutput` respectively must match, 152 `tf_device.replicate`, and associated `tf.TPUReplicatedInput` and 155 `tf.TPUReplicatedInput` and `tf.TPUReplicatedOutput` ops are simply forwarded to 165 %replicated_input = "tf.TPUReplicatedInput"(%arg0) : (tensor<i32>) -> tensor<i32> 189 …%replicated_input = "tf.TPUReplicatedInput"(%arg0, %arg1) : (tensor<i32>, tensor<i32>) -> tensor<i… 511 `tf.TPUReplicatedInput` (data parallelism) ops. It transforms a DAG where 513 `tf.TPUReplicatedInput` into a DAG where multiple `tf.TPUReplicatedInput` ops 525 %ri = "tf.TPUReplicatedInput"(%pi_0, %pi_1) : (!rtype, !rtype) -> !rtype 535 %ri_0 = "tf.TPUReplicatedInput"(%arg0, %arg2) : (!rtype, !rtype) -> !rtype [all …]
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
D | graph-function-defs.pbtxt | 47 op: "TPUReplicatedInput" 64 op: "TPUReplicatedInput"
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/executor_tpuv1_island_coarsening/ |
D | executor_tpuv1_island_coarsening.mlir | 100 // CHECK-NEXT: = "tf.TPUReplicatedInput" 102 …%outputs, %control = tf_executor.island wraps "tf.TPUReplicatedInput"(%arg0) {N = 1 : i64, T = i32…
|
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
D | legalization.md | 2948 ### tf.TPUReplicatedInput
|
/external/tensorflow/tensorflow/tools/api/golden/v1/ |
D | tensorflow.raw_ops.pbtxt | 4820 name: "TPUReplicatedInput"
|
/external/tensorflow/tensorflow/tools/api/golden/v2/ |
D | tensorflow.raw_ops.pbtxt | 4820 name: "TPUReplicatedInput"
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
D | tf_generated_ops.td | 15805 def TF_TPUReplicatedInputOp : TF_Op<"TPUReplicatedInput", [NoSideEffect]> { 15816 %replicated_input = "tf.TPUReplicatedInput"(%a, %b)
|
/external/tensorflow/tensorflow/go/op/ |
D | wrappers.go | 35087 func TPUReplicatedInput(scope *Scope, inputs []tf.Output, optional ...TPUReplicatedInputAttr) (outp… func
|
/external/tensorflow/tensorflow/core/ops/ |
D | ops.pbtxt | 54047 name: "TPUReplicatedInput"
|
/external/tensorflow/tensorflow/core/ops/compat/ |
D | ops_history.v2.pbtxt | 81568 name: "TPUReplicatedInput"
|