Searched refs:use_bn (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | model_subclassing_compiled_test.py | 47 num_hidden=32, num_classes=num_classes, use_dp=True, use_bn=True)
66 num_classes=num_classes, use_dp=True, use_bn=True)
88 num_hidden=32, num_classes=num_classes, use_dp=True, use_bn=True)
111 num_classes=num_classes, use_bn=True)
207 num_classes=num_classes, use_bn=True)
220 num_classes=num_classes, use_bn=True)
242 num_classes=num_classes, use_bn=True)
251 num_classes=num_classes, use_bn=True)
255 num_classes=num_classes, use_bn=True)
269 num_classes=num_classes, use_bn=True)
[all …]
|
| D | model_subclassing_test_util.py | 42 def get_multi_io_subclass_model(use_bn=False, use_dp=False, num_classes=(2, 3)): argument
51 if use_bn:
72 num_hidden=32, num_classes=4, use_bn=True, use_dp=True)
|
| D | model_subclassing_test.py | 138 num_hidden=32, num_classes=num_classes, use_dp=True, use_bn=True)
215 num_hidden=32, num_classes=num_classes, use_dp=True, use_bn=True)
232 num_hidden=32, num_classes=num_classes, use_dp=True, use_bn=True)
341 num_hidden=32, num_classes=4, use_bn=True, use_dp=True)
349 num_classes=(5, 6), use_bn=True, use_dp=True)
357 num_hidden=32, num_classes=4, use_bn=True, use_dp=True)
499 num_hidden=32, num_classes=num_classes, use_dp=True, use_bn=True)
515 num_classes=num_classes, use_dp=True, use_bn=True)
601 num_classes=num_classes, use_dp=True, use_bn=True)
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | testing_utils.py | 472 def __init__(self, num_hidden, num_classes, use_bn=False, use_dp=False): argument
474 self.use_bn = use_bn
482 if self.use_bn:
489 if self.use_bn:
|