Searched refs:module_inputs_torch_nn_TransformerEncoderLayer (Results 1 – 1 of 1) sorted by relevance
2402 for layer_module_input in module_inputs_torch_nn_TransformerEncoderLayer(2424 def module_inputs_torch_nn_TransformerEncoderLayer(module_info, device, dtype, requires_grad, train… function4133 module_inputs_func=module_inputs_torch_nn_TransformerEncoderLayer,