| 181 | } |
| 182 | |
| 183 | template <typename Store, typename TensorData> |
| 184 | NormWeights norm_data(Store & store, const TensorData & weight) { |
| 185 | return {tensor_data(store, weight), std::nullopt}; |
| 186 | } |
| 187 | |
| 188 | template <typename Store, typename TensorData> |
| 189 | NormWeights norm_data(Store & store, const TensorData & weight, const TensorData & bias) { |
| 190 | return {tensor_data(store, weight), tensor_data(store, bias)}; |
| 191 | } |
| 192 | |
| 193 | template <typename Store, typename WeightValues> |
| 194 | NormWeights norm( |
| 195 | Store & store, |
| 196 | const WeightValues & weight) { |
| 197 | return { |