Searched refs:running_averages (Results 1 – 2 of 2) sorted by relevance
16 bool running_averages = (in_running_mean != NULL); in dnn_batchnorm_op() local47 if (running_averages) { in dnn_batchnorm_op()98 running_averages ? running_average_factor : 0, in dnn_batchnorm_op()99 running_averages ? PyGpuArray_DEV_DATA(running_mean) : NULL, in dnn_batchnorm_op()100 running_averages ? PyGpuArray_DEV_DATA(running_var): NULL, in dnn_batchnorm_op()110 if (running_averages) { in dnn_batchnorm_op()
1819 self.running_averages = running_averages1826 if self.running_averages and self.inplace_running_mean:1828 if self.running_averages and self.inplace_running_var:1836 self.running_averages = False2793 if running_averages:2802 if running_averages:2806 batchnorm_op = GpuDnnBatchNorm(mode=mode, running_averages=running_averages)2807 if running_averages:3958 running_averages=node.op.running_averages,3969 running_averages=node.op.running_averages,[all …]