Searched refs:split_arg_into_blocks (Results 1 – 4 of 4) sorted by relevance
365 split_x = linear_operator_util.split_arg_into_blocks(532 split_rhs = linear_operator_util.split_arg_into_blocks(
469 split_x = linear_operator_util.split_arg_into_blocks(704 split_rhs = linear_operator_util.split_arg_into_blocks(
568 def split_arg_into_blocks(block_dims, block_dims_fn, arg, axis=-1): function
359 split_x = linear_operator_util.split_arg_into_blocks(577 split_rhs = linear_operator_util.split_arg_into_blocks(