Searched refs:num_new_per_replica_inputs (Results 1 – 1 of 1) sorted by relevance
852 const int num_new_per_replica_inputs = in RemoveUnusedXlaInput() local861 int dst_input = i * num_new_per_replica_inputs + iter->second; in RemoveUnusedXlaInput()882 int dst_input = num_replicas * num_new_per_replica_inputs + iter->second - in RemoveUnusedXlaInput()883 num_new_per_replica_inputs; in RemoveUnusedXlaInput()967 int new_variable_start_index = num_new_per_replica_inputs + in RemoveUnusedXlaInput()