Home
last modified time | relevance | path

Searched refs:broadcast_matrix_batch_dims (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/linalg/
Dlinear_operator_util_test.py98 linear_operator_util.broadcast_matrix_batch_dims([]))
102 tensor, = linear_operator_util.broadcast_matrix_batch_dims([arr])
117 x_bc, y_bc = linear_operator_util.broadcast_matrix_batch_dims([x, y])
135 x_bc, y_bc = linear_operator_util.broadcast_matrix_batch_dims([x, y])
156 x_bc, y_bc = linear_operator_util.broadcast_matrix_batch_dims([x_ph, y_ph])
175 x_bc, y_bc = linear_operator_util.broadcast_matrix_batch_dims([x_ph, y_ph])
186 linear_operator_util.broadcast_matrix_batch_dims([x, y])
189 linear_operator_util.broadcast_matrix_batch_dims([y, x])
Dlinear_operator_kronecker_test.py132 matrices = linear_operator_util.broadcast_matrix_batch_dims(matrices)
Dlinear_operator_block_diag_test.py124 matrices = linear_operator_util.broadcast_matrix_batch_dims(matrices)
Dlinear_operator_block_lower_triangular_test.py138 broadcasted_matrices = linear_operator_util.broadcast_matrix_batch_dims(
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_block_diag.py378 result_list = linear_operator_util.broadcast_matrix_batch_dims(
545 solution_list = linear_operator_util.broadcast_matrix_batch_dims(
617 diag_list = linear_operator_util.broadcast_matrix_batch_dims(diag_list)
631 broadcasted_blocks = linear_operator_util.broadcast_matrix_batch_dims(
671 eig_list = linear_operator_util.broadcast_matrix_batch_dims(eig_list)
Dlinear_operator_block_lower_triangular.py518 result_list = linear_operator_util.broadcast_matrix_batch_dims(
758 solution_list = linear_operator_util.broadcast_matrix_batch_dims(
830 diag_list = linear_operator_util.broadcast_matrix_batch_dims(diag_list)
843 flat_broadcast_operators = linear_operator_util.broadcast_matrix_batch_dims(
873 eig_list = linear_operator_util.broadcast_matrix_batch_dims(eig_list)
Dlinear_operator_util.py273 def broadcast_matrix_batch_dims(batch_matrices, name=None): function
378 matrix, rhs = broadcast_matrix_batch_dims([matrix, rhs])
Dlinear_operator_test_util.py369 split_matmul = linear_operator_util.broadcast_matrix_batch_dims(
586 split_solve = linear_operator_util.broadcast_matrix_batch_dims(
Dlinear_operator_circulant.py393 x, spec_mat = linear_operator_util.broadcast_matrix_batch_dims((x,