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.py407 split_x = linear_operator_util.split_arg_into_blocks(
593 split_rhs = linear_operator_util.split_arg_into_blocks(
Dlinear_operator_block_lower_triangular.py470 split_x = linear_operator_util.split_arg_into_blocks(
705 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.py376 split_x = linear_operator_util.split_arg_into_blocks(
602 split_rhs = linear_operator_util.split_arg_into_blocks(