diff options
Diffstat (limited to 'src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp')
| -rw-r--r-- | src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp b/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp index 394f970e54..b54389cf5f 100644 --- a/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp +++ b/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp @@ -62,9 +62,10 @@ Params extract_parameters(const ITensor *a, const ITensor *b, const ITensor *d, p.M = d->info()->tensor_shape().y(); p.K = a->info()->tensor_shape().x(); p.N = d->info()->tensor_shape().x(); + p.batches = 1; p.multis = 1; - p.indirect = false; p.sections = 1; + p.indirect = false; if(info.method == AsmConvMethod::Conv || info.method == AsmConvMethod::Indirect) { |