Searched refs:concat_along_batch_dimension (Results 1 – 1 of 1) sorted by relevance
1048 concat_along_batch_dimension(nest.flatten(nested_outs)))1152 def concat_along_batch_dimension(outputs): function