(self, grad, var)
| 170 | self._coll = collections |
| 171 | |
| 172 | def _mapper(self, grad, var): |
| 173 | name = var.op.name |
| 174 | if re.match('tower[0-9]+/', name): |
| 175 | # replicated training, var may come from different towers |
| 176 | return grad |
| 177 | if name not in SummaryGradient._summaried_gradient: |
| 178 | SummaryGradient._summaried_gradient.add(name) |
| 179 | tfv1.summary.histogram(name + '-grad', grad, collections=self._coll) |
| 180 | add_moving_summary(rms(grad, name=name + '/rms')) |
| 181 | return grad |
| 182 | |
| 183 | |
| 184 | class PrintGradient(MapGradient): |
nothing calls this directly
no test coverage detected