Searched refs:replicate_outputs (Results 1 – 2 of 2) sorted by relevance
1029 replicate_outputs = tpu.replicate(1039 if isinstance(replicate_outputs[0], list):1040 replicate_outputs = nest.flatten(replicate_outputs)1042 return replicate_outputs1052 replicate_outputs = rewrite_fn()1054 replicate_outputs = training_loop.repeat(iterations, rewrite_fn,1058 ctx.run_op = control_flow_ops.group(replicate_outputs)1060 if isinstance(replicate_outputs, list):1064 x for x in replicate_outputs if not isinstance(x, ops.Operation)1483 replicate_outputs = tpu.replicate([all …]
440 auto replicate_outputs = llvm::make_range( in ReplicateCluster() local450 use.set(*replicate_outputs.begin()); in ReplicateCluster()458 def->replaceAllUsesWith(replicate_outputs); in ReplicateCluster()