Searched refs:input_num_replicas (Results 1 – 1 of 1) sorted by relevance
2631 int input_num_replicas; in BuildTPUReplicateOps() local2633 GetNodeAttr(in_edges[pos]->src()->attrs(), "N", &input_num_replicas)); in BuildTPUReplicateOps()2651 if (!is_distributed_variable && input_num_replicas != num_replicas) { in BuildTPUReplicateOps()2655 "' has ", input_num_replicas, " replicas."); in BuildTPUReplicateOps()