| /external/tensorflow/tensorflow/python/ops/linalg/sparse/ |
| D | conjugate_gradient.py | 90 math_ops.matvec(
95 z = math_ops.matvec(operator, state.p)
102 q = preconditioner.matvec(r)
128 r0 = rhs - math_ops.matvec(operator, x)
132 p0 = math_ops.matvec(preconditioner, r0)
|
| /external/llvm-project/mlir/test/Dialect/Linalg/ |
| D | sparse_parallel.mlir | 126 // CHECK-PAR0-LABEL: func @matvec
131 // CHECK-PAR1-LABEL: func @matvec
136 // CHECK-PAR2-LABEL: func @matvec
141 // CHECK-PAR3-LABEL: func @matvec
146 // CHECK-PAR4-LABEL: func @matvec
151 func @matvec(%argA: tensor<16x32xf32>, %argb: tensor<32xf32>, %argx: tensor<16xf32>) -> tensor<16xf…
|
| D | tile.mlir | 180 func @matvec(%arg0: memref<?x?xf32, offset: ?, strides: [?, 1]>, %arg1: memref<?xf32, offset: ?, st…
181 linalg.matvec
187 // TILE-2-LABEL: func @matvec(
202 // TILE-2: linalg.matvec ins(%[[sAi]], %{{.*}} outs(%[[sCi]]
204 // TILE-02-LABEL: func @matvec(
219 // TILE-02: linalg.matvec ins(%[[sAj]], %[[sBj]]{{.*}} outs(%{{.*}}
221 // TILE-002-LABEL: func @matvec(
227 // TILE-234-LABEL: func @matvec(
250 // TILE-234: linalg.matvec ins(%[[sAij]], %[[sBj]]{{.*}} outs(%[[sCi]]
|
| D | transform-patterns.mlir | 35 func @matvec(%A: memref<?x?xf32, offset: ?, strides: [?, 1]>,
38 linalg.matvec
44 // CHECK-LABEL: func @matvec
50 // CHECK: linalg.matvec
162 linalg.matvec {__internal_linalg_transform__ = "__with_perm__"}
174 // CHECK: linalg.matvec
|
| /external/llvm-project/mlir/integration_test/Dialect/Linalg/CPU/ |
| D | matmul-vs-matvec.mlir | 21 func @matvec(%A: memref<?x?xf32>, %B: memref<?x?xf32>) -> (memref<?x?xf32>) {
33 linalg.matvec ins(%A, %b: memref<?x?xf32>, memref<?xf32, offset: ?, strides: [?]>)
53 %C2 = call @matvec(%A, %B) : (memref<?x?xf32>, memref<?x?xf32>) -> memref<?x?xf32>
59 assert %c, "Matmul does not produce same output as matvec"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.linalg.-linear-operator-block-diag.pbtxt | 154 name: "matvec"
155 …\', \'x\', \'adjoint\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'matvec\'], "
|
| D | tensorflow.linalg.-linear-operator-composition.pbtxt | 154 name: "matvec"
155 …\', \'x\', \'adjoint\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'matvec\'], "
|
| D | tensorflow.linalg.-linear-operator-permutation.pbtxt | 154 name: "matvec"
155 …\', \'x\', \'adjoint\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'matvec\'], "
|
| D | tensorflow.linalg.-linear-operator-inversion.pbtxt | 154 name: "matvec"
155 …\', \'x\', \'adjoint\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'matvec\'], "
|
| D | tensorflow.linalg.-linear-operator-toeplitz.pbtxt | 158 name: "matvec"
159 …\', \'x\', \'adjoint\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'matvec\'], "
|
| D | tensorflow.linalg.-linear-operator-identity.pbtxt | 151 name: "matvec"
152 …\', \'x\', \'adjoint\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'matvec\'], "
|
| D | tensorflow.linalg.-linear-operator-kronecker.pbtxt | 154 name: "matvec"
155 …\', \'x\', \'adjoint\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'matvec\'], "
|
| D | tensorflow.linalg.-linear-operator-tridiag.pbtxt | 158 name: "matvec"
159 …\', \'x\', \'adjoint\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'matvec\'], "
|
| D | tensorflow.linalg.-linear-operator-full-matrix.pbtxt | 150 name: "matvec"
151 …\', \'x\', \'adjoint\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'matvec\'], "
|
| D | tensorflow.linalg.-linear-operator.pbtxt | 149 name: "matvec"
150 …\', \'x\', \'adjoint\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'matvec\'], "
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.linalg.-linear-operator-block-lower-triangular.pbtxt | 154 name: "matvec"
155 …\', \'x\', \'adjoint\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'matvec\'], "
|
| D | tensorflow.linalg.-linear-operator-block-diag.pbtxt | 154 name: "matvec"
155 …\', \'x\', \'adjoint\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'matvec\'], "
|
| D | tensorflow.linalg.-linear-operator-toeplitz.pbtxt | 158 name: "matvec"
159 …\', \'x\', \'adjoint\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'matvec\'], "
|
| D | tensorflow.linalg.-linear-operator-diag.pbtxt | 154 name: "matvec"
155 …\', \'x\', \'adjoint\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'matvec\'], "
|
| D | tensorflow.linalg.-linear-operator-permutation.pbtxt | 154 name: "matvec"
155 …\', \'x\', \'adjoint\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'matvec\'], "
|
| D | tensorflow.linalg.-linear-operator-householder.pbtxt | 154 name: "matvec"
155 …\', \'x\', \'adjoint\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'matvec\'], "
|
| D | tensorflow.linalg.-linear-operator-kronecker.pbtxt | 154 name: "matvec"
155 …\', \'x\', \'adjoint\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'matvec\'], "
|
| D | tensorflow.linalg.-linear-operator-full-matrix.pbtxt | 150 name: "matvec"
151 …\', \'x\', \'adjoint\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'matvec\'], "
|
| D | tensorflow.linalg.-linear-operator-inversion.pbtxt | 154 name: "matvec"
155 …\', \'x\', \'adjoint\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'matvec\'], "
|
| D | tensorflow.linalg.-linear-operator.pbtxt | 149 name: "matvec"
150 …\', \'x\', \'adjoint\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'matvec\'], "
|