Searched refs:tpu_replicated_input (Results 1 – 5 of 5) sorted by relevance
417 auto tpu_replicated_input = llvm::cast<TF::TPUReplicatedInputOp>(input); in ReplicateCluster() local418 int64_t tpu_replicated_input_index = tpu_replicated_input.index(); in ReplicateCluster()431 if (tpu_replicated_input.is_mirrored_variable()) in ReplicateCluster()
104 return tpu_ops.tpu_replicated_input(inputs=tensors)
789 Node* tpu_replicated_input = nullptr; in CreateConcatAndSplitNodesForInputTensor() local806 .Finalize(graph, &tpu_replicated_input)); in CreateConcatAndSplitNodesForInputTensor()808 << tpu_replicated_input->DebugString(); in CreateConcatAndSplitNodesForInputTensor()813 .Input(tpu_replicated_input) in CreateConcatAndSplitNodesForInputTensor()824 output_from_concat_node = tpu_replicated_input; in CreateConcatAndSplitNodesForInputTensor()
379 handle = tpu_ops.tpu_replicated_input(replicated_vars,1397 tpu_ops.tpu_replicated_input(
596 self._replica_id = tpu_ops.tpu_replicated_input(