Searched defs:input_shard_axes (Results 1 – 1 of 1) sorted by relevance
1088 input_shard_axes=None, argument1238 input_shard_axes=None, argument