diff options
Diffstat (limited to 'src/cpu')
| -rw-r--r-- | src/cpu/operators/CpuMatMul.cpp | 1 | ||||
| -rw-r--r-- | src/cpu/operators/CpuMatMul.h | 12 |
2 files changed, 7 insertions, 6 deletions
diff --git a/src/cpu/operators/CpuMatMul.cpp b/src/cpu/operators/CpuMatMul.cpp index 64b5167ad0..87cb6c6b54 100644 --- a/src/cpu/operators/CpuMatMul.cpp +++ b/src/cpu/operators/CpuMatMul.cpp @@ -191,6 +191,7 @@ void CpuMatMul::configure(ITensorInfo *lhs, ITensorInfo *rhs, ITensorInfo *dst, // Fill AsmGemmInfo class object before configuration _gemm_info.activation_info = info.fused_activation(); _gemm_info.fast_mode = settings.fast_math(); + _gemm_info.negated_offsets = false; lhs_to_use = (_adj_lhs) ? _lhs_transposed : lhs_to_use; rhs_to_use = (_adj_rhs) ? _rhs_transposed : rhs_to_use; diff --git a/src/cpu/operators/CpuMatMul.h b/src/cpu/operators/CpuMatMul.h index ae6345141e..9f5833b24f 100644 --- a/src/cpu/operators/CpuMatMul.h +++ b/src/cpu/operators/CpuMatMul.h @@ -21,8 +21,8 @@ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE * SOFTWARE. */ -#ifndef SRC_CPU_OPERATORS_CPUMATMUL -#define SRC_CPU_OPERATORS_CPUMATMUL +#ifndef ACL_SRC_CPU_OPERATORS_CPUMATMUL +#define ACL_SRC_CPU_OPERATORS_CPUMATMUL #include "arm_compute/core/TensorInfo.h" #include "src/core/common/Macros.h" @@ -59,9 +59,9 @@ public: * Note: Check documentation of @ref NEMatMul for a list of supported datatypes and layouts * * - * @param[in] lhs Source tensor info. - * @param[in] rhs Source tensor info. - * @param[out] dst Destination tensor info. Data types supported: same as @p lhs / @p rhs. + * @param[in] lhs Left-hand side tensor info. + * @param[in] rhs Right-hand side tensor info. + * @param[out] dst Output tensor to store the result of the batched matrix multiplication. Data types supported: same as @p lhs / @p rhs. * @param[in] info Contains MatMul operation information described in @ref MatMulInfo. * @param[in] settings The settings for matmul operation (i.e fast math) */ @@ -112,4 +112,4 @@ private: } } -#endif /* SRC_CPU_OPERATORS_CPUMATMUL */ +#endif /* ACL_SRC_CPU_OPERATORS_CPUMATMUL */ |