Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A 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)
H A Dsparse_grad.py173 b_grad = array_ops.matrix_transpose(b_grad, conjugate=True)
H A Dinit_ops.py958 q = array_ops.matrix_transpose(q)
1100 return math_ops.matmul(c, array_ops.matrix_transpose(c))
H A Dinit_ops_v2.py700 q = array_ops.matrix_transpose(q)
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/linalg/
H A 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)
H A 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)
/aosp_15_r20/external/swiftshader/third_party/llvm-16.0/llvm/lib/Transforms/Scalar/
H A DLowerMatrixIntrinsics.cpp580 case Intrinsic::matrix_transpose: in supportsShapeInfo()
616 } else if (match(Inst, m_Intrinsic<Intrinsic::matrix_transpose>( in propagateShapeForward()
692 } else if (match(V, m_Intrinsic<Intrinsic::matrix_transpose>( in propagateShapeBackward()
771 if (!match(&I, m_Intrinsic<Intrinsic::matrix_transpose>( in sinkTranspose()
777 if (match(TA, m_Intrinsic<Intrinsic::matrix_transpose>(m_Value(TATA)))) { in sinkTranspose()
872 match(A, m_Intrinsic<Intrinsic::matrix_transpose>(m_Value(AT))) && in liftTranspose()
873 match(B, m_Intrinsic<Intrinsic::matrix_transpose>(m_Value((BT))))) { in liftTranspose()
886 match(A, m_Intrinsic<Intrinsic::matrix_transpose>( in liftTranspose()
888 match(B, m_Intrinsic<Intrinsic::matrix_transpose>( in liftTranspose()
937 case Intrinsic::matrix_transpose: in Visit()
[all …]
/aosp_15_r20/external/swiftshader/third_party/llvm-10.0/llvm/lib/Transforms/Scalar/
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A 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\']…
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/
H A 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()))
H A 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]
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A 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()
/aosp_15_r20/external/bazelbuild-rules_rust/examples/ffi/rust_calling_c/c/
H A Dmatrix.h40 void matrix_transpose(Matrix* matrix);
H A Dmatrix_test.c71 matrix_transpose(matrix); in test_transpose()
H A Dmatrix.c57 void matrix_transpose(Matrix* matrix) { in matrix_transpose() function
/aosp_15_r20/external/bazelbuild-rules_rust/examples/ffi/rust_calling_c/src/
H A Dffi.rs30 pub fn matrix_transpose(matrix: *mut Matrix); in matrix_transpose() function
H A Dmatrix.rs92 ffi::matrix_transpose(self.matrix); in transpose()
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.linalg.pbtxt208 name: "matrix_transpose"
209 … \'name\', \'conjugate\'], varargs=None, keywords=None, defaults=[\'matrix_transpose\', \'False\']…
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
H A 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(
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/
H A 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.
/aosp_15_r20/external/swiftshader/third_party/llvm-16.0/llvm/include/llvm/IR/
H A DMatrixBuilder.h119 getModule(), Intrinsic::matrix_transpose, OverloadedTypes);
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567b/include/llvm/IR/
DMatrixBuilder.h119 getModule(), Intrinsic::matrix_transpose, OverloadedTypes);
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567/include/llvm/IR/
DMatrixBuilder.h119 getModule(), Intrinsic::matrix_transpose, OverloadedTypes);
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r522817/include/llvm/IR/
DMatrixBuilder.h119 getModule(), Intrinsic::matrix_transpose, OverloadedTypes);
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r536225/include/llvm/IR/
DMatrixBuilder.h119 getModule(), Intrinsic::matrix_transpose, OverloadedTypes);

12