Searched refs:TPUPartitionedOutput (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | tpu_resource_partitioning.mlir | 17 // CHECK: [[OUTPUT:%.+]]:2 = "tf.TPUPartitionedOutput"([[COMPUTATION]])
38 // CHECK-NOT: tf.TPUPartitionedOutput
64 // CHECK-NOT: tf.TPUPartitionedOutput
74 // CHECK-NOT: tf.TPUPartitionedOutput
103 // CHECK-NOT: tf.TPUPartitionedOutput
114 // CHECK-NOT: tf.TPUPartitionedOutput
|
| D | tpu_sharding_identification.mlir | 269 …%2 = "tf.TPUPartitionedOutput"(%1#1) {_XlaSharding = "\04\05\06", partition_dim = -1 : i64} : (ten…
313 %2 = "tf.TPUPartitionedOutput"(%1) {partition_dim = -1 : i64} : (tensor<*xi32>) -> tensor<*xi32>
|
| D | tpu_rewrite.mlir | 1215 // input/output using `tf.TPUPartitionedInput` and `tf.TPUPartitionedOutput`.
1237 // CHECK-NOT: tf.TPUPartitionedOutput
1238 …%partitioned_output:2 = "tf.TPUPartitionedOutput"(%computation) {N = 2 : i64, partition_dim = -1 :…
1253 // output using `tf.TPUPartitionedInput` and `tf.TPUPartitionedOutput`.
1275 // CHECK-NOT: tf.TPUPartitionedOutput
1276 …%partitioned_output:2 = "tf.TPUPartitionedOutput"(%computation) {_XlaSharding = "\08\03\1A\02\01\0…
1301 …%partitioned_output:2 = "tf.TPUPartitionedOutput"(%computation) {N = 2 : i64, partition_dim = -1 :…
1324 …%partitioned_output:2 = "tf.TPUPartitionedOutput"(%computation) {N = 2 : i64, partition_dim = -1 :…
1345 …// expected-error@+1 {{'tf.TPUPartitionedOutput' op must be a unique user of tf_device.cluster_fun…
1346 …%partitioned_output:2 = "tf.TPUPartitionedOutput"(%computation) {N = 2 : i64, partition_dim = -1 :…
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | TPUPartitionedOutput.pbtxt | 2 name: "TPUPartitionedOutput"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_TPUPartitionedOutput.pbtxt | 2 graph_op_name: "TPUPartitionedOutput"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tf_passes.td | 581 …%partitioned_output:2 = "tf.TPUPartitionedOutput"(%computation) {N = 2 : i64, _XlaSharding = "", p…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops.td | 2090 def TF_TPUPartitionedOutputOp : TF_Op<"TPUPartitionedOutput", [NoSideEffect]> {
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 49766 func TPUPartitionedOutput(scope *Scope, inputs tf.Output, num_splits int64, optional ...TPUPartitio… func
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 53936 name: "TPUPartitionedOutput"
|