Home
last modified time | relevance | path

Searched defs:num_per_replica_inputs (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc787 int num_per_replica_inputs = in RemoveUnusedXlaInput() local
1251 int num_per_replica_inputs = in ReplaceArgUsedByOutsideCompilationWithPlaceholder() local
Ddistributed_tpu_rewrite_pass.cc3988 const int num_per_replica_inputs = input_types.size() - num_distributed_vars; in GetIOTypes() local