Searched defs:output_shard_axes (Results 1 – 1 of 1) sorted by relevance
1090 output_shard_axes=None, argument1240 output_shard_axes=None, argument