split_arg_into_blocks (reference) in projects: aosp_15_r20

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/linalg/
H A D	linear_operator_block_diag.py	`395 split_x = linear_operator_util.split_arg_into_blocks( 581 split_rhs = linear_operator_util.split_arg_into_blocks(`
H A D	linear_operator_block_lower_triangular.py	`474 split_x = linear_operator_util.split_arg_into_blocks( 709 split_rhs = linear_operator_util.split_arg_into_blocks(`
H A D	linear_operator_util.py	`565 def split_arg_into_blocks(block_dims, block_dims_fn, arg, axis=-1): function`
H A D	linear_operator_test_util.py	`467 split_x = linear_operator_util.split_arg_into_blocks( 693 split_rhs = linear_operator_util.split_arg_into_blocks(`