Home
last modified time | relevance | path

Searched refs:TPUPartitionedInput (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtpu_resource_partitioning.mlir10 // CHECK: [[INPUT:%.+]] = "tf.TPUPartitionedInput"([[READ0]], [[READ1]])
13 …%0 = "tf.TPUPartitionedInput"(%arg0, %arg1) {N = 2 : i64, _XlaSharding = "", partition_dim = -1 : …
31 // CHECK: [[INPUT:%.+]] = "tf.TPUPartitionedInput"([[READ0]], [[READ1]])
34 …%0 = "tf.TPUPartitionedInput"(%arg0, %arg1) {N = 2 : i64, _XlaSharding = "", partition_dim = -1 : …
48 // CHECK: "tf.TPUPartitionedInput"([[ARG0]], [[ARG1]])
49 …%0 = "tf.TPUPartitionedInput"(%arg0, %arg1) {N = 2 : i64, _XlaSharding = "", partition_dim = -1 : …
52 // CHECK: "tf.TPUPartitionedInput"([[ARG0]], [[ARG1]])
53 …%3 = "tf.TPUPartitionedInput"(%arg0, %arg1) {N = 2 : i64, _XlaSharding = "", partition_dim = -1 : …
61 // CHECK-NOT: tf.TPUPartitionedInput
71 // CHECK-NOT: tf.TPUPartitionedInput
[all …]
Dtpu_reorder_replicate_and_partitioned_inputs.mlir8 // CHECK: [[PI:%.*]] = "tf.TPUPartitionedInput"([[RI_0]], [[RI_1]])
9 …%pi_0 = "tf.TPUPartitionedInput"(%arg0, %arg1) {_XlaSharding = "", partition_dim = -1 : i64} : (te…
10 …%pi_1 = "tf.TPUPartitionedInput"(%arg2, %arg3) {_XlaSharding = "", partition_dim = -1 : i64} : (te…
21 // CHECK: [[PI:%.*]] = "tf.TPUPartitionedInput"([[RI_0]], [[RI_1]])
22 …%pi_0 = "tf.TPUPartitionedInput"(%arg0, %arg1) {device = "", partition_dim = -1 : i64} : (tensor<!…
23 …%pi_1 = "tf.TPUPartitionedInput"(%arg2, %arg3) {device = "", partition_dim = -1 : i64} : (tensor<!…
30 // a tf.TPUPartitionedInput op.
35 // CHECK: [[PI_0:%.*]] = "tf.TPUPartitionedInput"([[ARG0]], [[ARG1]])
36 …%pi_0 = "tf.TPUPartitionedInput"(%arg0, %arg1) {device = "", partition_dim = -1 : i64} : (tensor<!…
37 // CHECK: [[PI_1:%.*]] = "tf.TPUPartitionedInput"([[ARG2]], [[ARG3]])
[all …]
Dtpu_sharding_identification.mlir264 …%0 = "tf.TPUPartitionedInput"(%arg0) {_XlaSharding = "\01\02\03", partition_dim = -1 : i64} : (ten…
286 …%0 = "tf.TPUPartitionedInput"(%arg0) {_XlaSharding = "\01\02\03", partition_dim = -1 : i64} : (ten…
308 %0 = "tf.TPUPartitionedInput"(%arg0) {partition_dim = -1 : i64} : (tensor<*xi32>) -> tensor<*xi32>
331 …%0 = "tf.TPUPartitionedInput"(%arg0) {_XlaSharding = "\01\02\03", partition_dim = -1 : i64} : (ten…
Dtpu_cluster_formation.mlir256 // TPUPartitionedInput nodes would be inside the replicate but outside the
269 …%4 = "tf.TPUPartitionedInput"(%2, %3) {_XlaSharding = "", device = "", partition_dim = -1 : i64} :…
287 // CHECK: %[[PI:[0-9]*]] = "tf.TPUPartitionedInput"(%[[RI_0]], %[[RI_1]])
606 // expected-error@+1 {{'tf.TPUPartitionedInput' op requires 4 operands but found 2}}
607 …%4 = "tf.TPUPartitionedInput"(%2, %3) {_XlaSharding = "", device = "", partition_dim = -1 : i64} :…
Dtpu_rewrite.mlir1215 // input/output using `tf.TPUPartitionedInput` and `tf.TPUPartitionedOutput`.
1223 // CHECK-NOT: tf.TPUPartitionedInput
1224 …%partitioned_input = "tf.TPUPartitionedInput"(%read0, %read1) {N = 2 : i64, partition_dim = -1 : i…
1253 // output using `tf.TPUPartitionedInput` and `tf.TPUPartitionedOutput`.
1261 // CHECK-NOT: tf.TPUPartitionedInput
1262 …%partitioned_input = "tf.TPUPartitionedInput"(%read0, %read1) {_XlaSharding = "\08\03\1A\02\01\02\…
1297 …%partitioned_input = "tf.TPUPartitionedInput"(%read0, %read1) {N = 2 : i64, partition_dim = -1 : i…
1320 …%partitioned_input = "tf.TPUPartitionedInput"(%read0, %read1) {N = 2 : i64, partition_dim = -1 : i…
1343 …%partitioned_input = "tf.TPUPartitionedInput"(%read0, %read1) {N = 2 : i64, partition_dim = -1 : i…
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DTPUPartitionedInput.pbtxt2 name: "TPUPartitionedInput"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_TPUPartitionedInput.pbtxt2 graph_op_name: "TPUPartitionedInput"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtf_passes.td510 inputs. It reorders `tf.TPUPartitionedInput` (model parallelism) and
512 multiple `tf.TPUPartitionedInput` ops are feeding into a single
514 are feeding into a single `tf.TPUPartitionedInput`. Transforming the IR in such
523 …%pi_0 = "tf.TPUPartitionedInput"(%arg0, %arg1) {_XlaSharding = "", device = "", partition_dim = -1…
524 …%pi_1 = "tf.TPUPartitionedInput"(%arg2, %arg3) {_XlaSharding = "", device = "", partition_dim = -1…
537 …%pi = "tf.TPUPartitionedInput"(%ri_0, %ri_1) {_XlaSharding = "", device = "", partition_dim = -1 :…
551 resource variable (from `tf.TPUPartitionedInput`) to individual partitioned
552 resource variables (`tf.TPUPartitionedInput` operands). As resource op
561 …%partitioned_variable = "tf.TPUPartitionedInput"(%arg0, %arg1) {N = 2 : i64, _XlaSharding = "", pa…
579 …%partitioned_input = "tf.TPUPartitionedInput"(%read0, %read1) {N = 2 : i64, _XlaSharding = "", par…
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops.td2070 def TF_TPUPartitionedInputOp : TF_Op<"TPUPartitionedInput", [NoSideEffect]> {
/external/tensorflow/tensorflow/go/op/
Dwrappers.go18043 func TPUPartitionedInput(scope *Scope, inputs []tf.Output, optional ...TPUPartitionedInputAttr) (ou… func
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt53907 name: "TPUPartitionedInput"