Searched defs:num_replicated_inputs (Results 1 – 2 of 2) sorted by relevance
344 int32_t num_replicated_inputs = replicated_inputs.size() * n; in ParseReplicateOp() local381 const int32_t num_replicated_inputs = in Print() local451 const int32_t num_replicated_inputs = in Verify() local545 int32_t num_replicated_inputs = replicated_inputs.size() * n; in BuildReplicateOp() local
937 int num_replicated_inputs = num_per_replica_inputs + num_distributed_vars; in RemoveUnusedXlaInput() local2670 const int num_replicated_inputs = distributed_var_start_index; in BuildTPUReplicateOps() local