Searched refs:get_static_batch_dim (Results 1 – 1 of 1) sorted by relevance
501 def get_static_batch_dim(output_shape): function507 get_static_batch_dim(ts._to_legacy_output_shapes()) # pylint: disable=protected-access