Searched refs:num_params_per_layer (Results 1 – 3 of 3) sorted by relevance
482 num_params_per_layer = CUDNN_LSTM_PARAMS_PER_LAYER484 num_params_per_layer = CUDNN_GRU_PARAMS_PER_LAYER486 num_params_per_layer = CUDNN_RNN_TANH_PARAMS_PER_LAYER488 num_params_per_layer = CUDNN_RNN_RELU_PARAMS_PER_LAYER490 num_params_per_layer *= num_dirs491 self.assertEqual(num_params_per_layer * num_layers, len(lhs))494 layer_lhs = lhs[i * num_params_per_layer: (i+1) * num_params_per_layer]495 layer_rhs = rhs[i * num_params_per_layer: (i+1) * num_params_per_layer]
937 num_params_per_layer = CUDNN_LSTM_PARAMS_PER_LAYER939 num_params_per_layer = CUDNN_GRU_PARAMS_PER_LAYER941 num_params_per_layer = CUDNN_RNN_RELU_PARAMS_PER_LAYER943 num_params_per_layer = CUDNN_RNN_TANH_PARAMS_PER_LAYER946 num_params = num_layers * num_params_per_layer
1112 const int num_params_per_layer = num_params_ / num_layers / num_dirs; in Compute() local1115 const int num_params_input_state = num_params_per_layer / 2; in Compute()1118 CHECK(num_params_per_layer % 2 == 0) in Compute()1127 const int layer_idx = i / num_params_per_layer; in Compute()1128 const int index_within_layer = i % num_params_per_layer; in Compute()