Home
last modified time | relevance | path

Searched refs:matrix_transpose (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_util.py440 a = array_ops.matrix_transpose(a, conjugate=True)
442 a = array_ops.matrix_transpose(a, conjugate=False)
444 b = array_ops.matrix_transpose(b, conjugate=True)
446 b = array_ops.matrix_transpose(b, conjugate=False)
Dlinalg_impl.py61 transpose = array_ops.matrix_transpose
125 return array_ops.matrix_transpose(matrix, conjugate=True)
658 rhs = array_ops.matrix_transpose(rhs, conjugate=conjugate_rhs)
Dlinear_operator_test_util.py1377 embedded = array_ops.matrix_transpose(embedded_t)
/external/tensorflow/tensorflow/python/ops/
Dlinalg_grad.py431 sqrtm_transpose = array_ops.matrix_transpose(sqrtm)
440 vec_da = array_ops.reshape(array_ops.matrix_transpose(grad), shape_vec_da)
447 return array_ops.matrix_transpose(dsqrtm_transpose)
599 zx_sym = xzt + array_ops.matrix_transpose(xzt)
911 gv1t = array_ops.matrix_transpose(grad_v1, conjugate=True)
937 grad_a = array_ops.matrix_transpose(
965 superdiag_conj = array_ops.matrix_transpose(op.inputs[0], conjugate=True)
966 maindiag_conj = array_ops.matrix_transpose(op.inputs[1], conjugate=True)
967 subdiag_conj = array_ops.matrix_transpose(op.inputs[2], conjugate=True)
Dsparse_grad.py173 b_grad = array_ops.matrix_transpose(b_grad, conjugate=True)
Dinit_ops.py958 q = array_ops.matrix_transpose(q)
1100 return math_ops.matmul(c, array_ops.matrix_transpose(c))
Dinit_ops_v2.py700 q = array_ops.matrix_transpose(q)
Darray_ops.py2380 def matrix_transpose(a, name="matrix_transpose", conjugate=False): function
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.linalg.pbtxt196 name: "matrix_transpose"
197 … \'name\', \'conjugate\'], varargs=None, keywords=None, defaults=[\'matrix_transpose\', \'False\']…
257 … \'name\', \'conjugate\'], varargs=None, keywords=None, defaults=[\'matrix_transpose\', \'False\']…
Dtensorflow.pbtxt1684 name: "matrix_transpose"
1685 … \'name\', \'conjugate\'], varargs=None, keywords=None, defaults=[\'matrix_transpose\', \'False\']…
/external/swiftshader/third_party/llvm-10.0/llvm/lib/Transforms/Scalar/
DLowerMatrixIntrinsics.cpp310 case Intrinsic::matrix_transpose: in supportsShapeInfo()
347 } else if (match(Inst, m_Intrinsic<Intrinsic::matrix_transpose>( in propagateShapeForward()
424 } else if (match(V, m_Intrinsic<Intrinsic::matrix_transpose>( in propagateShapeBackward()
474 case Intrinsic::matrix_transpose: in Visit()
545 case Intrinsic::matrix_transpose: in VisitCallInst()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster_test.cc390 const Eigen::array<Eigen::DenseIndex, 2> matrix_transpose({1, 0}); in TEST_F() local
416 VLOG(1) << "x = [" << x_matrix.shuffle(matrix_transpose) << "] y = [" in TEST_F()
417 << y_matrix.shuffle(matrix_transpose) << "] lambda = " << lambda; in TEST_F()
/external/tensorflow/tensorflow/python/kernel_tests/linalg/
Dcholesky_op_test.py45 grad_a += math_ops.conj(array_ops.matrix_transpose(grad_a))
64 z_h = math_ops.conj(array_ops.matrix_transpose(l_inverse_middle))
270 x, math_ops.conj(array_ops.matrix_transpose(x))) / shape[0]
Dlinear_operator_circulant_test.py819 matrix_t = array_ops.matrix_transpose(matrix_tensor)
821 matrix, matrix_transpose, imag_matrix = self.evaluate(
825 self.assertAllClose(matrix, matrix_transpose, atol=1e-6)
1004 array_ops.matrix_transpose(operator.to_dense())
1021 array_ops.matrix_transpose(math_ops.conj(operator_imag.to_dense()))
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.linalg.pbtxt208 name: "matrix_transpose"
209 … \'name\', \'conjugate\'], varargs=None, keywords=None, defaults=[\'matrix_transpose\', \'False\']…
/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Darray_ops_test.py60 transposed = array_ops.matrix_transpose(matrix)
68 transposed = array_ops.matrix_transpose(matrix, conjugate=True)
79 transposed = array_ops.matrix_transpose(batch_matrix)
94 return array_ops.matrix_transpose(matrix)
113 return array_ops.matrix_transpose(matrix)
120 array_ops.matrix_transpose(vector)
136 transposed = array_ops.matrix_transpose(
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dmath_test.py679 math_ops.matmul(z, array_ops.matrix_transpose(z)) # Ensure pos. def.
759 x = z + array_ops.matrix_transpose(z) # Ensure self-adjoint.
/external/tensorflow/tensorflow/python/keras/initializers/
Dinitializers_v2.py611 q = array_ops.matrix_transpose(q)
/external/swiftshader/third_party/llvm-10.0/configs/common/include/llvm/IR/
DIntrinsicEnums.inc180 matrix_transpose, // llvm.matrix.transpose
/external/tensorflow/tensorflow/python/ops/distributions/
Dutil.py81 matrix_t = array_ops.matrix_transpose(matrix)