Searched refs:get_small_functional_mlp (Results 1 – 3 of 3) sorted by relevance
300 def get_small_functional_mlp(num_hidden, num_classes, input_dim): function360 return get_small_functional_mlp(num_hidden, num_classes, input_dim)
234 model = testing_utils.get_small_functional_mlp(1, 4, input_dim=3)
390 model = testing_utils.get_small_functional_mlp(1, 4, input_dim=3)