diff options
Diffstat (limited to 'arm_compute/runtime')
-rw-r--r-- | arm_compute/runtime/NEON/functions/NEGEMMConvolutionLayer.h | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEGEMMConvolutionLayer.h b/arm_compute/runtime/NEON/functions/NEGEMMConvolutionLayer.h index 2af11ad656..a28266265d 100644 --- a/arm_compute/runtime/NEON/functions/NEGEMMConvolutionLayer.h +++ b/arm_compute/runtime/NEON/functions/NEGEMMConvolutionLayer.h @@ -131,7 +131,7 @@ public: * * The user can query the database of optimised kernels in * arm_gemm by specifying one of the enumerations of - * arm_gemm::WeightFormat in the weight_format field of the input + * arm_compute::WeightFormat in the weight_format field of the input * parameter weights_info. In case of success, the method * writes the expected format in the output parameter * expected_weight_format. The expected_weight_format can than be @@ -140,7 +140,7 @@ public: * * Use case one - query for a specific format: * - * WeightInfo weights_info(..., arm_gemm::WeightFormat::OHWIo4, ...); // Set the value of the input query. + * WeightInfo weights_info(..., arm_compute::WeightFormat::OHWIo4, ...); // Set the value of the input query. * if (NEGEMMConvolutionlayer::has_opt_impl(WeightFormat(), ...., weights_info, ...)) * { * auto conv = std::unique_ptr<NEGEMMConvolutionlayer>(); @@ -150,8 +150,8 @@ public: * * Use case two - query for any format that would be optimal for the GEMM to execute: * - * WeightInfo weights_info(..., arm_gemm::WeightFormat::ANY, ...); // Set the value of the input query. - * arm_gemm::WeightFormat expected_wf; + * WeightInfo weights_info(..., arm_compute::WeightFormat::ANY, ...); // Set the value of the input query. + * arm_compute::WeightFormat expected_wf; * if (NEGEMMConvolutionlayer::has_opt_impl(expected_wf, ...., weights_info, ...)) * { * auto conv = std::unique_ptr<NEGEMMConvolutionlayer>(); @@ -177,7 +177,7 @@ public: * * @return a Status */ - static Status has_opt_impl(arm_gemm::WeightFormat &expected_weight_format, const ITensorInfo *src, const ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *dst, + static Status has_opt_impl(arm_compute::WeightFormat &expected_weight_format, const ITensorInfo *src, const ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *dst, const PadStrideInfo &conv_info, const WeightsInfo &weights_info = WeightsInfo(), const Size2D &dilation = Size2D(1U, 1U), const ActivationLayerInfo &act_info = ActivationLayerInfo(), bool enable_fast_math = false); |