Searched refs:_cross_replica_concat (Results 1 – 1 of 1) sorted by relevance
291 def _cross_replica_concat(tensor, core_id, num_cores, name): function1005 _cross_replica_concat(1018 _cross_replica_concat(