Searched refs:arg_split_dim (Results 1 – 1 of 1) sorted by relevance
520 def arg_is_blockwise(block_dimensions, arg, arg_split_dim): argument531 x).shape[arg_split_dim] for x in arg]