Searched refs:get_small_functional_mlp (Results 1 – 3 of 3) sorted by relevance
411 def get_small_functional_mlp(num_hidden, num_classes, input_dim): function482 return get_small_functional_mlp(num_hidden, num_classes, input_dim)
238 model = testing_utils.get_small_functional_mlp(1, 4, input_dim=3)
340 model = testing_utils.get_small_functional_mlp(1, 4, input_dim=3)