(self, histograms, ret='tensor')
cached cur layer histograms
cached cur layer histograms
cached cur layer histograms
def _cached_histograms(self, histograms, ret='tensor'):
        """
        cached cur layer histograms
        """
        if ret == 'tb':
            self._cur_layer_dtables.append(histograms)
        elif ret == 'tensor':
            result_nid, result = histograms
            for node_id, result_hist in zip(result_nid, result):
                self._cur_layer_cached_histograms[node_id] = result_hist
def _cached_histograms ( self , histograms , ret = 'tensor' ) : if ret == 'tb' : self . _cur_layer_dtables . append ( histograms ) elif ret == 'tensor' : result_nid , result = histograms for node_id , result_hist in zip ( result_nid , result ) : self . _cur_layer_cached_histograms [ node_id ] = result_hist