Searched refs:matrix_multiply (Results 1 – 13 of 13) sorted by relevance
| /external/python/pybind11/tests/ |
| D | test_eigen.py | 693 assert np.all(m.matrix_multiply(a, b) == np.array([[3.0], [7], [11]]))
694 assert np.all(m.matrix_multiply(A=a, B=b) == np.array([[3.0], [7], [11]]))
695 assert np.all(m.matrix_multiply(B=b, A=a) == np.array([[3.0], [7], [11]]))
698 m.matrix_multiply(b, a)
702 m.matrix_multiply(A=b, B=a)
706 m.matrix_multiply(B=a, A=b)
|
| /external/llvm-project/mlir/test/Dialect/Linalg/ |
| D | codegen-strategy.mlir | 11 // CHECK: vector.matrix_multiply
|
| /external/swiftshader/third_party/llvm-10.0/llvm/lib/Transforms/Scalar/ |
| D | LowerMatrixIntrinsics.cpp | 309 case Intrinsic::matrix_multiply: in supportsShapeInfo()
343 if (match(Inst, m_Intrinsic<Intrinsic::matrix_multiply>( in propagateShapeForward()
415 if (match(V, m_Intrinsic<Intrinsic::matrix_multiply>( in propagateShapeBackward()
473 case Intrinsic::matrix_multiply: in Visit()
542 case Intrinsic::matrix_multiply: in VisitCallInst()
|
| /external/llvm-project/mlir/integration_test/Dialect/Vector/CPU/ |
| D | test-matrix-multiply-row.mlir | 42 %c = vector.matrix_multiply %a, %b
|
| D | test-matrix-multiply-col.mlir | 42 %c = vector.matrix_multiply %a, %b
|
| /external/llvm-project/llvm/lib/Transforms/Scalar/ |
| D | LowerMatrixIntrinsics.cpp | 512 case Intrinsic::matrix_multiply: in supportsShapeInfo()
546 if (match(Inst, m_Intrinsic<Intrinsic::matrix_multiply>( in propagateShapeForward()
618 if (match(V, m_Intrinsic<Intrinsic::matrix_multiply>( in propagateShapeBackward()
676 case Intrinsic::matrix_multiply: in Visit()
704 if (match(&I, m_Intrinsic<Intrinsic::matrix_multiply>())) in Visit()
759 case Intrinsic::matrix_multiply: in VisitCallInst()
1607 case Intrinsic::matrix_multiply: in writeFnName()
1636 case Intrinsic::matrix_multiply: in getNumShapeArgs()
|
| /external/mesa3d/src/compiler/spirv/ |
| D | vtn_alu.c | 63 matrix_multiply(struct vtn_builder *b, in matrix_multiply() function
203 return matrix_multiply(b, vtn_ssa_transpose(b, src1), src0); in vtn_handle_matrix_alu()
205 return matrix_multiply(b, src0, src1); in vtn_handle_matrix_alu()
|
| /external/llvm-project/llvm/include/llvm/IR/ |
| D | MatrixBuilder.h | 144 getModule(), Intrinsic::matrix_multiply, OverloadedTypes);
|
| /external/swiftshader/third_party/llvm-10.0/configs/common/include/llvm/IR/ |
| D | IntrinsicEnums.inc | 179 matrix_multiply, // llvm.matrix.multiply
|
| /external/llvm-project/mlir/include/mlir/Dialect/Vector/ |
| D | VectorOps.td | 1994 def Vector_MatmulOp : Vector_Op<"matrix_multiply", [NoSideEffect,
2017 into 'vector.matrix_multiply' operations to bring the operations closer
2020 The ‘vector.matrix_multiply’ op treats `lhs` as matrix with <lhs_rows> rows
2032 %C = vector.matrix_multiply %A, %B
|
| /external/llvm-project/mlir/test/Dialect/Vector/ |
| D | vector-contract-transforms.mlir | 524 // MATRIX: %[[mm1:.*]] = vector.matrix_multiply %[[a3]], %[[b7]] {lhs_columns = 4 : i32, lhs_…
|
| /external/llvm-project/llvm/lib/IR/ |
| D | Verifier.cpp | 5068 case Intrinsic::matrix_multiply: in visitIntrinsicCall()
5080 case Intrinsic::matrix_multiply: in visitIntrinsicCall()
|
| /external/llvm-project/mlir/test/Conversion/VectorToLLVM/ |
| D | vector-to-llvm.mlir | 833 %C = vector.matrix_multiply %A, %B
|