Searched refs:swapaxes (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | svd_op_test.py | 44 return np.matmul(u * s[..., None, :], np.swapaxes(v, -1, -2))
56 u_diff = np.matmul(u_val, np.swapaxes(u_val, -1, -2)) - np.eye(m)
57 v_diff = np.matmul(v_val, np.swapaxes(v_val, -1, -2)) - np.eye(n)
|
| D | self_adjoint_eig_op_test.py | 38 x_np = x_np + np.swapaxes(x_np, -1, -2)
48 v_diff = np.matmul(v_val, np.swapaxes(v_val, -1, -2)) - np.eye(n)
|
| D | matrix_solve_op_test.py | 37 np_ans = np.linalg.solve(np.swapaxes(a, -2, -1) if adjoint else a, b)
|
| D | matrix_triangular_solve_op_test.py | 74 transp = lambda x: np.swapaxes(x, -1, -2)
|
| D | fft_test.py | 93 expected = np.swapaxes(expected, -1, -2)
|
| /external/libvpx/libvpx/tools/non_greedy_mv/ |
| D | non_greedy_mv.py | 62 plane = np.swapaxes(np.swapaxes(plane, 0, 1), 1, 2)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | batch_matmul_op_test.py | 52 x = np.conjugate(np.swapaxes(x, -1, -2))
54 y = np.conjugate(np.swapaxes(y, -1, -2))
|
| D | svd_op_test.py | 191 CompareSingularVectors(self, np.conj(np.swapaxes(v_np, -2, -1)), v_tf_val,
|
| D | rnn_cell_test.py | 1534 out = np.swapaxes(out, 0, 1)
|
| /external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
| D | stats_ops_test.py | 1348 expected_stats_summary = np.swapaxes(expected_stats_summary, 1, 2)
1402 expected_stats_summary = np.swapaxes(expected_stats_summary, 1, 2)
1427 expected_stats_summary = np.swapaxes(expected_stats_summary, 1, 2)
1488 expected_stats_summary = np.swapaxes(expected_stats_summary, 1, 2)
1517 expected_stats_summary = np.swapaxes(expected_stats_summary, 1, 2)
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_array_ops_test.py | 1051 self.assertAllEqual([[1], [2], [3]], np_array_ops.swapaxes(x, 0, 1))
1052 self.assertAllEqual([[1], [2], [3]], np_array_ops.swapaxes(x, -2, -1))
1055 np_array_ops.swapaxes(x, 0, 2))
1057 np_array_ops.swapaxes(x, -3, -1))
|
| D | np_array_ops.py | 798 def swapaxes(a, axis1, axis2): # pylint: disable=missing-docstring function
1043 return swapaxes(arrays, 0, axis)
1432 swapaxes_ = lambda t: swapaxes(t, axis, -1)
|
| /external/tensorflow/third_party/py/numpy/tf_numpy_api/ |
| D | tensorflow.experimental.numpy.pbtxt | 828 name: "swapaxes"
|