Searched refs:num_distributed_vars (Results 1 – 2 of 2) sorted by relevance
454 int num_distributed_vars; in MoveHeadOutsideCompilationToHost() local456 &num_distributed_vars)); in MoveHeadOutsideCompilationToHost()461 (input_types.size() - num_distributed_vars) / num_replicas; in MoveHeadOutsideCompilationToHost()543 int num_other_inputs = num_distributed_vars + num_variables + in MoveHeadOutsideCompilationToHost()553 num_distributed_vars); in MoveHeadOutsideCompilationToHost()564 for (int i = input_types.size() - num_distributed_vars; in MoveHeadOutsideCompilationToHost()575 num_new_per_replica_input_types / num_replicas + num_distributed_vars + in MoveHeadOutsideCompilationToHost()718 TF_RET_CHECK(index >= new_num_per_replica_inputs + num_distributed_vars); in MoveHeadOutsideCompilationToHost()784 int num_distributed_vars; in RemoveUnusedXlaInput() local786 &num_distributed_vars)); in RemoveUnusedXlaInput()[all …]
3984 int num_distributed_vars; in GetIOTypes() local3987 &num_distributed_vars)); in GetIOTypes()3988 const int num_per_replica_inputs = input_types.size() - num_distributed_vars; in GetIOTypes()4012 num_replicas, num_per_replica_inputs / num_replicas, num_distributed_vars, in GetIOTypes()