NN parameter更新の論理問題

AとBは　NN(Neural Network)の一部とします(前後関係は下図)
A--->B---> loss

lossでAのparameterをのみ更新したい場合(Bは更新しない),
Bを冷凍(gradの計算を不能に)したらだめですね。

この分野に明るい方、ご教授お願い致します。