diff options
Diffstat (limited to 'arm_compute/runtime')
-rw-r--r-- | arm_compute/runtime/GLES_COMPUTE/functions/GCFullyConnectedLayer.h | 1 | ||||
-rw-r--r-- | arm_compute/runtime/NEON/functions/assembly/NEGEMMInterleavedWrapper.h | 1 |
2 files changed, 2 insertions, 0 deletions
diff --git a/arm_compute/runtime/GLES_COMPUTE/functions/GCFullyConnectedLayer.h b/arm_compute/runtime/GLES_COMPUTE/functions/GCFullyConnectedLayer.h index 3a13e659f9..64e98e2f4d 100644 --- a/arm_compute/runtime/GLES_COMPUTE/functions/GCFullyConnectedLayer.h +++ b/arm_compute/runtime/GLES_COMPUTE/functions/GCFullyConnectedLayer.h @@ -94,6 +94,7 @@ private: void configure_conv_fc(const IGCTensor *input, const IGCTensor *weights, IGCTensor *output); MemoryGroup _memory_group; + IWeightsManager *_weights_manager; GCIm2ColKernel _im2col_kernel; GCFullyConnectedLayerReshapeWeights _reshape_weights_kernel; GCGEMMMatrixMultiplyKernel _mm_kernel; diff --git a/arm_compute/runtime/NEON/functions/assembly/NEGEMMInterleavedWrapper.h b/arm_compute/runtime/NEON/functions/assembly/NEGEMMInterleavedWrapper.h index eeea0babf1..695dcd5b6e 100644 --- a/arm_compute/runtime/NEON/functions/assembly/NEGEMMInterleavedWrapper.h +++ b/arm_compute/runtime/NEON/functions/assembly/NEGEMMInterleavedWrapper.h @@ -120,6 +120,7 @@ public: private: MemoryGroup _memory_group; + IWeightsManager *_weights_manager; bool _is_prepared{ false }; bool _pretranspose_b{ false }; Window _block_walker{}; |