Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_util.py125 def broadcast_matrix_batch_dims(batch_matrices, name=None): function
223 chol, rhs = broadcast_matrix_batch_dims([chol, rhs])
327 a, b = broadcast_matrix_batch_dims([a, b])
353 matrix, rhs = broadcast_matrix_batch_dims([matrix, rhs])
401 matrix, rhs = broadcast_matrix_batch_dims([matrix, rhs])
Dlinear_operator_block_diag.py268 result_list = linear_operator_util.broadcast_matrix_batch_dims(
294 solution_list = linear_operator_util.broadcast_matrix_batch_dims(
303 diag_list = linear_operator_util.broadcast_matrix_batch_dims(diag_list)
317 broadcasted_blocks = linear_operator_util.broadcast_matrix_batch_dims(
Dlinear_operator_circulant.py391 x, spec_mat = linear_operator_util.broadcast_matrix_batch_dims((x,
/external/tensorflow/tensorflow/python/kernel_tests/linalg/
Dlinear_operator_util_test.py102 linear_operator_util.broadcast_matrix_batch_dims([]))
106 tensor, = linear_operator_util.broadcast_matrix_batch_dims([arr])
123 x_bc, y_bc = linear_operator_util.broadcast_matrix_batch_dims([x, y])
142 x_bc, y_bc = linear_operator_util.broadcast_matrix_batch_dims([x, y])
165 x_bc, y_bc = linear_operator_util.broadcast_matrix_batch_dims([x_ph, y_ph])
186 x_bc, y_bc = linear_operator_util.broadcast_matrix_batch_dims([x_ph, y_ph])
198 linear_operator_util.broadcast_matrix_batch_dims([x, y])
201 linear_operator_util.broadcast_matrix_batch_dims([y, x])
Dlinear_operator_block_diag_test.py116 matrices = linear_operator_util.broadcast_matrix_batch_dims(matrices)
Dlinear_operator_kronecker_test.py133 matrices = linear_operator_util.broadcast_matrix_batch_dims(matrices)