diff options
Diffstat (limited to 'arm_compute/runtime/GLES_COMPUTE/functions')
-rw-r--r-- | arm_compute/runtime/GLES_COMPUTE/functions/GCFullyConnectedLayer.h | 15 |
1 files changed, 6 insertions, 9 deletions
diff --git a/arm_compute/runtime/GLES_COMPUTE/functions/GCFullyConnectedLayer.h b/arm_compute/runtime/GLES_COMPUTE/functions/GCFullyConnectedLayer.h index cd108c3eab..63565df1a7 100644 --- a/arm_compute/runtime/GLES_COMPUTE/functions/GCFullyConnectedLayer.h +++ b/arm_compute/runtime/GLES_COMPUTE/functions/GCFullyConnectedLayer.h @@ -75,17 +75,14 @@ public: GCFullyConnectedLayer &operator=(GCFullyConnectedLayer &&) = default; /** Set the input and output tensors. * - * @param[in] input Source tensor. Data type supported: F16/F32. - * @param[in] weights Weights tensor. The weights must be 2 dimensional. Data type supported: Same as @p input - * @param[in] biases Bias tensor. It can be nullptr. Data type supported:Same as @p input. - * @param[out] output Destination tensor. Data type supported: Same as @p input. - * @param[in] transpose_weights (Optional) Transpose weights if true. Defaults to true. - * @param[in] are_weights_reshaped (Optional) Reshape the weights tensor if false. Defaults to false. - * @param[in] retain_internal_weights (Optional) Retain internal reshaped weights. Defaults to false. - * Used for reconfiguration purposes. + * @param[in] input Source tensor. Data type supported: F16/F32. + * @param[in] weights Weights tensor. The weights must be 2 dimensional. Data type supported: Same as @p input + * @param[in] biases Bias tensor. It can be nullptr. Data type supported:Same as @p input. + * @param[out] output Destination tensor. Data type supported: Same as @p input. + * @param[in] fc_info (Optional) Fully connected layer additional info */ void configure(const IGCTensor *input, const IGCTensor *weights, const IGCTensor *biases, IGCTensor *output, - bool transpose_weights = true, bool are_weights_reshaped = false, bool retain_internal_weights = false); + FullyConnectedLayerInfo fc_info = FullyConnectedLayerInfo()); //Inherited methods override void run() override; |