Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dencapsulate_tpu_computations_pass.cc428 int num_distributed_vars; in MoveHeadOutsideCompilationToHost() local
753 int num_distributed_vars; in RemoveUnusedXlaInput() local
1208 int num_distributed_vars; in ReplaceArgUsedByOutsideCompilationWithPlaceholder() local
1446 int num_distributed_vars; in RemoveEdgesBetweenArgAndRetval() local
2623 const int num_distributed_vars = pos - num_replicated_inputs; in BuildTPUReplicateOps() local
H A Ddistributed_tpu_rewrite_pass.cc4439 int num_distributed_vars; in GetIOTypes() local