Searched refs:_num_params_per_layer (Results 1 – 2 of 2) sorted by relevance
153 _num_params_per_layer = None variable in _CudnnRNN461 num_gates = self._num_params_per_layer // 2478 return [[self._num_units]] * num_dirs * self._num_params_per_layer544 _num_params_per_layer = CUDNN_LSTM_PARAMS_PER_LAYER variable in CudnnLSTM590 _num_params_per_layer = CUDNN_GRU_PARAMS_PER_LAYER variable in CudnnGRU597 _num_params_per_layer = CUDNN_RNN_TANH_PARAMS_PER_LAYER variable in CudnnRNNTanh604 _num_params_per_layer = CUDNN_RNN_RELU_PARAMS_PER_LAYER variable in CudnnRNNRelu
213 self._num_params_per_layer * self._num_layers * self._num_dirs)293 layer_weights_num = self._num_params_per_layer * self._num_dirs406 _num_params_per_layer = CUDNN_LSTM_PARAMS_PER_LAYER variable in CudnnParamsFormatConverterLSTM497 _num_params_per_layer = CUDNN_GRU_PARAMS_PER_LAYER variable in CudnnParamsFormatConverterGRU612 _num_params_per_layer = CUDNN_RNN_TANH_PARAMS_PER_LAYER variable in CudnnParamsFormatConverterTanh618 _num_params_per_layer = CUDNN_RNN_RELU_PARAMS_PER_LAYER variable in CudnnParamsFormatConverterRelu