Home
last modified time | relevance | path

Searched refs:num_distributed_vars (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc454 int num_distributed_vars; in MoveHeadOutsideCompilationToHost() local
456 &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() local
786 &num_distributed_vars)); in RemoveUnusedXlaInput()
[all …]
Ddistributed_tpu_rewrite_pass.cc3984 int num_distributed_vars; in GetIOTypes() local
3987 &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()