Searched refs:get_small_functional_mlp (Results 1 – 3 of 3) sorted by relevance
461 def get_small_functional_mlp(num_hidden, num_classes, input_dim): function532 return get_small_functional_mlp(num_hidden, num_classes, input_dim)
222 model = testing_utils.get_small_functional_mlp(1, 4, input_dim=3)
328 model = testing_utils.get_small_functional_mlp(1, 4, input_dim=3)