Searched defs:first_layer_args (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/benchmarks/fastrnns/ |
| D | custom_lstms.py | 285 def init_stacked_lstm(num_layers, layer, first_layer_args, other_layer_args): argument
295 def __init__(self, num_layers, layer, first_layer_args, other_layer_args): argument
325 def __init__(self, num_layers, layer, first_layer_args, other_layer_args): argument
353 def __init__(self, num_layers, layer, first_layer_args, other_layer_args): argument
|