Searched defs:replicated_input (Results 1 – 2 of 2) sorted by relevance
34 TF::TPUReplicatedInputOp replicated_input) { in ReorderReplicateAndPartitionedInputs()110 getFunction()->walk([](TF::TPUReplicatedInputOp replicated_input) { in runOnFunction()
292 const auto& replicated_input = replicated_input_and_idx.value(); in SetReplicateOpOperands() local529 for (auto& replicated_input : replicated_inputs) { in BuildReplicateOp() local