Home
last modified time | relevance | path

Searched refs:split_arg_into_blocks (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_block_diag.py365 split_x = linear_operator_util.split_arg_into_blocks(
532 split_rhs = linear_operator_util.split_arg_into_blocks(
Dlinear_operator_block_lower_triangular.py469 split_x = linear_operator_util.split_arg_into_blocks(
704 split_rhs = linear_operator_util.split_arg_into_blocks(
Dlinear_operator_util.py568 def split_arg_into_blocks(block_dims, block_dims_fn, arg, axis=-1): function
Dlinear_operator_test_util.py359 split_x = linear_operator_util.split_arg_into_blocks(
577 split_rhs = linear_operator_util.split_arg_into_blocks(