mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-06-27 23:11:57 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #24344 from asmorkalov:as/einsum_fallback_fix

Browse Source
This commit is contained in:
Alexander Alekhin 2023-09-29 16:48:41 +00:00
commit f3c724d449
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
modules/dnn/src/layers/einsum_layer.cpp
View File

@ -38,7 +38,6 @@ Mat batchwiseMatMul(
const Mat& input2, const Mat& input2,
const MatShape& input2ShapeOverride) const MatShape& input2ShapeOverride)
{ {
// Sanity checks before the actual MatMul // Sanity checks before the actual MatMul
//input_1.DataType() == input_2.DataType(), "Data types of the inputs must match for MatMul"); //input_1.DataType() == input_2.DataType(), "Data types of the inputs must match for MatMul");
@ -391,6 +390,15 @@ public:
OutputArrayOfArrays outputs_arr, OutputArrayOfArrays outputs_arr,
OutputArrayOfArrays internals_arr) CV_OVERRIDE OutputArrayOfArrays internals_arr) CV_OVERRIDE
{ {
CV_TRACE_FUNCTION();
CV_TRACE_ARG_VALUE(name, "name", name.c_str());
if (inputs_arr.depth() == CV_16S)
{
forward_fallback(inputs_arr, outputs_arr, internals_arr);
return;
}
// homogenize inputs // homogenize inputs
preProcessInputs(inputs_arr); preProcessInputs(inputs_arr);