Home
last modified time | relevance | path

Searched refs:matrix_multiply (Results 1 – 7 of 7) sorted by relevance

/external/ComputeLibrary/tests/validation/reference/
DWinograd.cpp341 matrix_multiply<T>(matrix, src_tile, tmp_tile); in winograd_input_transform()
342 matrix_multiply<T>(tmp_tile, matrix_transposed, dst_tile); in winograd_input_transform()
419 matrix_multiply<T>(trans_matrix, input_tile, tmp_tile); in winograd_filter_transform()
422 matrix_multiply<T>(tmp_tile, trans_matrix_transposed, transf_tile); in winograd_filter_transform()
544 matrix_multiply<T>(trans_matrix, input_tile, tmp_tile); in winograd_output_transform()
547 matrix_multiply<T>(tmp_tile, trans_matrix_transposed, output_tile); in winograd_output_transform()
/external/python/pybind11/tests/
Dtest_eigen.py693 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/ComputeLibrary/tests/validation/
DHelpers.cpp163 void matrix_multiply(const SimpleTensor<T> &a, const SimpleTensor<T> &b, SimpleTensor<T> &out) in matrix_multiply() function
390 template void matrix_multiply(const SimpleTensor<float> &a, const SimpleTensor<float> &b, SimpleTen…
391 template void matrix_multiply(const SimpleTensor<half> &a, const SimpleTensor<half> &b, SimpleTenso…
DHelpers.h177 void matrix_multiply(const SimpleTensor<T> &a, const SimpleTensor<T> &b, SimpleTensor<T> &out);
/external/swiftshader/third_party/llvm-10.0/llvm/lib/Transforms/Scalar/
DLowerMatrixIntrinsics.cpp309 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/mesa3d/src/compiler/spirv/
Dvtn_alu.c63 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/swiftshader/third_party/llvm-10.0/configs/common/include/llvm/IR/
DIntrinsicEnums.inc179 matrix_multiply, // llvm.matrix.multiply