diff options
Diffstat (limited to 'src/runtime')
-rw-r--r-- | src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp | 2 | ||||
-rw-r--r-- | src/runtime/cpu/operators/CpuDepthwiseConvolutionAssemblyDispatch.cpp | 1 |
2 files changed, 1 insertions, 2 deletions
diff --git a/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp b/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp index 7fb1d583ff..c58a662f10 100644 --- a/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp +++ b/src/runtime/NEON/functions/NEGEMMAssemblyDispatch.cpp @@ -75,7 +75,7 @@ Params extract_parameters(const ITensor *a, const ITensor *b, const ITensor *d, else { p.multis = b->info()->tensor_shape().z(); - p.batches = d->info()->tensor_shape().total_size_upper(2) / p.multis; //COMPMID-1423: Agree on and document the layout of gemm inputs/outputs + p.batches = d->info()->tensor_shape().total_size_upper(2) / p.multis; } // Update M in case of GEMM3D for output diff --git a/src/runtime/cpu/operators/CpuDepthwiseConvolutionAssemblyDispatch.cpp b/src/runtime/cpu/operators/CpuDepthwiseConvolutionAssemblyDispatch.cpp index 5f5304cded..039714abb1 100644 --- a/src/runtime/cpu/operators/CpuDepthwiseConvolutionAssemblyDispatch.cpp +++ b/src/runtime/cpu/operators/CpuDepthwiseConvolutionAssemblyDispatch.cpp @@ -459,7 +459,6 @@ bool CpuDepthwiseConvolutionAssemblyDispatch::is_optimized_supported(const ITens } // Check data type - // TODO (COMPMID-3004): Add assembly optimized routine for QASYMM8_SIGNED NEDepthwiseConvolutionLayer const DataType input_type = input->data_type(); const bool is_input_type_valid = is_data_type_float(input_type) || input_type == DataType::QASYMM8; const DataType weights_type = weights->data_type(); |