- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for TPUPartitionedInput (0.44 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
inputs. It reorders `tf.TPUPartitionedInput` (model parallelism) and `tf.TPUReplicatedInput` (data parallelism) ops. It transforms a DAG where multiple `tf.TPUPartitionedInput` ops are feeding into a single `tf.TPUReplicatedInput` into a DAG where multiple `tf.TPUReplicatedInput` ops are feeding into a single `tf.TPUPartitionedInput`. Transforming the IR in such
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0)