Searched refs:tpu_replicated_input (Results 1 – 4 of 4) sorted by relevance
403 auto tpu_replicated_input = llvm::cast<TF::TPUReplicatedInputOp>(input); in ReplicateCluster() local404 int64_t tpu_replicated_input_index = tpu_replicated_input.index(); in ReplicateCluster()415 if (tpu_replicated_input.is_mirrored_variable()) in ReplicateCluster()
94 return tpu_ops.tpu_replicated_input(inputs=tensors)
374 handle = tpu_ops.tpu_replicated_input([v.handle for v in replicated_vars],1381 tpu_ops.tpu_replicated_input(
596 self._replica_id = tpu_ops.tpu_replicated_input(