Searched refs:num_replicas_node (Results 1 – 1 of 1) sorted by relevance
599 auto num_replicas_node = graph_utils::AddScalarConstNode(num_replicas, graph); in RewriteRebatchV2ToV1() local600 rebatch_node->add_input(num_replicas_node->name()); in RewriteRebatchV2ToV1()