Searched refs:matmul_with_broadcast (Results 1 – 6 of 6) sorted by relevance
357 uh_x = linear_operator_util.matmul_with_broadcast(360 v_d_uh_x = linear_operator_util.matmul_with_broadcast(364 vh_x = linear_operator_util.matmul_with_broadcast(367 u_d_vh_x = linear_operator_util.matmul_with_broadcast(u, d_vh_x)428 vh_linv_rhs = linear_operator_util.matmul_with_broadcast(438 u_capinv_vh_linv_rhs = linear_operator_util.matmul_with_broadcast(453 vh_linv_u = linear_operator_util.matmul_with_broadcast(
179 return linear_operator_util.matmul_with_broadcast(
194 return linear_operator_util.matmul_with_broadcast(
227 def matmul_with_broadcast(a, function
268 mat_matmul = linear_operator_util.matmul_with_broadcast(
255 result = linear_operator_util.matmul_with_broadcast(x, y)273 result = linear_operator_util.matmul_with_broadcast(x, y)291 result = linear_operator_util.matmul_with_broadcast(314 result = linear_operator_util.matmul_with_broadcast(339 linear_operator_util.matmul_with_broadcast(x_ph, y_ph),