25 #ifndef ARM_COMPUTE_IWEIGHTSMANAGER_H 26 #define ARM_COMPUTE_IWEIGHTSMANAGER_H 81 std::map<const ITensor *, std::vector<ITransformWeights *>> _managed_weights;
82 std::map<const ITensor *, ITransformWeights *> _managed_weights_parents;
void manage(const ITensor *weights, ITransformWeights *parent=nullptr)
Start managing a weights tensor.
IWeightsManager()
Constructor.
Interface for Neon tensor.
Copyright (c) 2017-2021 Arm Limited.
bool are_weights_managed(const ITensor *weights)
Check if the weights are managed.
virtual ~IWeightsManager()=default
Default Destructor.
Weights manager interface to handle weights transformations.
IWeightsManager & operator=(const IWeightsManager &)=delete
Prevent instances of this class to be copied.
ITensor * run(const ITensor *weights, ITransformWeights *weights_transform)
Run the reshape function.
ITensor * acquire(const ITensor *weights, ITransformWeights *weights_transform)
Acquire the requested reshape tensor of the selected weights.