Searched refs:weights_aux (Results 1 – 2 of 2) sorted by relevance
221 …ITensor *weights_aux = utils::cast::polymorphic_cast<ITensor *>(tensors.get_tensor(offset_in… in prepare() local222 ARM_COMPUTE_ERROR_ON_NULLPTR(weights, weights_aux); in prepare()224 CpuAuxTensorHandler permuted_weights(_perm_weights, *weights_aux); in prepare()
379 …ITensor *weights_aux = utils::cast::polymorphic_cast<ITensor *>(tensors.get_tensor(offset_in… in prepare() local381 CpuAuxTensorHandler permuted_weights(_weights_hwio, *weights_aux); in prepare()