Searched defs:replicated_input (Results 1 – 3 of 3) sorted by relevance
34 TF::TPUReplicatedInputOp replicated_input) { in ReorderReplicateAndPartitionedInputs()110 getFunction()->walk([](TF::TPUReplicatedInputOp replicated_input) { in runOnFunction()
296 const auto& replicated_input = replicated_input_and_idx.value(); in SetReplicateOpOperands() local533 for (auto& replicated_input : replicated_inputs) { in BuildReplicateOp() local
792 std::vector<NodeBuilder::NodeOut> replicated_input; in CreateConcatAndSplitNodesForInputTensor() local