Searched refs:_cudnn_to_tf_weights (Results 1 – 1 of 1) sorted by relevance
386 def _cudnn_to_tf_weights(self, *cu_weights): member in CudnnParamsFormatConverter416 def _cudnn_to_tf_weights(self, *cu_weights): member in CudnnParamsFormatConverterLSTM487 (w,) = self._cudnn_to_tf_weights(*cu_weights)501 def _cudnn_to_tf_weights(self, *cu_weights): member in CudnnParamsFormatConverterGRU559 W_ir, w_h, r_h = self._cudnn_to_tf_weights(*cu_weights)569 def _cudnn_to_tf_weights(self, *cu_weights): member in CudnnParamsFormatConverterBasic