Searched refs:block_dimensions (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_block_diag.py | 337 block_dimensions = (self._block_range_dimensions() if adjoint
339 if linear_operator_util.arg_is_blockwise(block_dimensions, x, arg_dim):
344 block_dimensions[i].assert_is_compatible_with(block.shape[arg_dim])
356 block_dimensions = (self._block_range_dimensions() if adjoint
359 block_dimensions, x, arg_dim)
411 block_dimensions = (self._block_range_dimensions() if adjoint
413 if linear_operator_util.arg_is_blockwise(block_dimensions, x, -1):
418 block_dimensions[i].assert_is_compatible_with(block.shape[-1])
510 block_dimensions = (self._block_domain_dimensions() if adjoint
514 block_dimensions, rhs, arg_dim)
[all …]
|
| D | linear_operator_block_lower_triangular.py | 441 block_dimensions = (self._block_range_dimensions() if adjoint
443 if linear_operator_util.arg_is_blockwise(block_dimensions, x, arg_dim):
448 block_dimensions[i].assert_is_compatible_with(block.shape[arg_dim])
460 block_dimensions = (self._block_range_dimensions() if adjoint
463 block_dimensions, x, arg_dim)
551 block_dimensions = (self._block_range_dimensions() if adjoint
553 if linear_operator_util.arg_is_blockwise(block_dimensions, x, -1):
558 block_dimensions[i].assert_is_compatible_with(block.shape[-1])
679 block_dimensions = (self._block_domain_dimensions() if adjoint
683 block_dimensions, rhs, arg_dim)
[all …]
|
| D | linear_operator_util.py | 520 def arg_is_blockwise(block_dimensions, arg, arg_split_dim): argument
524 if (isinstance(arg, (tuple, list)) and len(arg) == len(block_dimensions)):
532 self_dims = [dim.value for dim in block_dimensions]
|