Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtpu_cluster_formation.cc446 llvm::SmallVector<Operation*, 8> ordered_tpu_replicate_inputs = in ReplicateCluster() local
448 ordered_tpu_replicate_inputs.append(packed_ops.begin(), packed_ops.end()); in ReplicateCluster()
452 llvm::enumerate(ordered_tpu_replicate_inputs)) { in ReplicateCluster()
501 llvm::zip(ordered_tpu_replicate_inputs, in ReplicateCluster()