Searched defs:num_replicated_inputs (Results 1 – 2 of 2) sorted by relevance
355 int32_t num_replicated_inputs = replicated_inputs.size() * n; in parse() local388 const int32_t num_replicated_inputs = in print() local464 int32_t num_replicated_inputs = replicated_inputs.size() * n; in BuildReplicateOp() local505 const int32_t num_replicated_inputs = in verify() local
906 int num_replicated_inputs = num_per_replica_inputs + num_distributed_vars; in RemoveUnusedXlaInput() local2622 const int num_replicated_inputs = distributed_var_start_index; in BuildTPUReplicateOps() local