Home
last modified time | relevance | path

Searched refs:TPUReplicatedInput (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtpu_cluster_formation.mlir181 // 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 …]
Dtpu_reorder_replicate_and_partitioned_inputs.mlir6 // 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 …]
Dgraph_pruning_preserve_ops.mlir1 …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…
Dshape_inference.mlir178 // 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/
DTPUReplicatedInput.pbtxt2 name: "TPUReplicatedInput"
24 name: "TPUReplicatedInput"
53 name: "TPUReplicatedInput"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DTPUReplicatedInput.pbtxt2 name: "TPUReplicatedInput"
24 name: "TPUReplicatedInput"
60 name: "TPUReplicatedInput"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_TPUReplicatedInput.pbtxt2 graph_op_name: "TPUReplicatedInput"
13 %replicated_input = "tf.TPUReplicatedInput"(%a, %b)
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tpu_bridge_v1/
Dend_to_end.mlir14 …%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/
Dtf_passes.td138 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/
Dgraph-function-defs.pbtxt47 op: "TPUReplicatedInput"
64 op: "TPUReplicatedInput"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/executor_tpuv1_island_coarsening/
Dexecutor_tpuv1_island_coarsening.mlir100 // 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/
Dlegalization.md2948 ### tf.TPUReplicatedInput
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt4820 name: "TPUReplicatedInput"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt4820 name: "TPUReplicatedInput"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td15805 def TF_TPUReplicatedInputOp : TF_Op<"TPUReplicatedInput", [NoSideEffect]> {
15816 %replicated_input = "tf.TPUReplicatedInput"(%a, %b)
/external/tensorflow/tensorflow/go/op/
Dwrappers.go35087 func TPUReplicatedInput(scope *Scope, inputs []tf.Output, optional ...TPUReplicatedInputAttr) (outp… func
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt54047 name: "TPUReplicatedInput"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt81568 name: "TPUReplicatedInput"