| /external/tensorflow/tensorflow/python/keras/benchmarks/layer_benchmarks/ |
| D | layer_benchmarks_test.py | 38 def _get_layer_args(layer_cls, layer_args): argument
40 if layer_cls is tf.keras.layers.Bidirectional:
217 def benchmark_layer_call(self, layer_cls, layer_args, inputs, num_iters): argument
218 layer = layer_cls(**_get_layer_args(layer_cls, layer_args))
228 self, layer_cls, layer_args, inputs, num_iters): argument
229 layer = layer_cls(**_get_layer_args(layer_cls, layer_args))
240 self, layer_cls, layer_args, inputs, num_iters): argument
243 if layer_cls is tf.keras.layers.Embedding and name[-1] == "GPU":
245 layer = layer_cls(**_get_layer_args(layer_cls, layer_args))
256 self, layer_cls, layer_args, inputs, num_iters): argument
[all …]
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | testing_utils.py | 92 def layer_test(layer_cls, argument
169 layer = layer_cls(**kwargs)
176 (layer_cls.__name__, layer.supports_masking, supports_masking, kwargs))
187 if 'weights' in tf_inspect.getargspec(layer_cls.__init__):
189 layer = layer_cls(**kwargs)
197 (layer_cls.__name__, x, backend.dtype(y),
206 (layer_cls.__name__, x, actual, expected, kwargs))
217 (layer_cls.__name__, x, actual, expected, kwargs))
238 (layer_cls.__name__, x, actual_output.dtype,
292 (layer_cls.__name__,
|
| /external/scapy/scapy/modules/krack/ |
| D | automaton.py | 203 def build_ap_info_pkt(self, layer_cls, dest): argument
210 / layer_cls(timestamp=0, beacon_interval=100,
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | recurrent_test.py | 642 def test_rnn_cell_trainability(self, layer_cls): argument
644 layer = layer_cls(3, trainable=False)
1717 def test_for_enable_caching_device_for_layer(self, layer_cls): argument
1719 layer = layer_cls(1)
1727 layer = layer_cls(1, enable_caching_device=non_default_value)
|
| D | convolutional_test.py | 460 def test_group_conv(self, layer_cls, input_shape): argument
465 layer = layer_cls(16, 3, groups=4, use_bias=False)
|
| D | gru_v2_test.py | 297 def build_model(layer_cls): argument
300 layer = layer_cls(rnn_state_size,
|
| D | lstm_v2_test.py | 451 def build_model(layer_cls): argument
454 layer = layer_cls(rnn_state_size,
|
| /external/tensorflow/tensorflow/python/keras/saving/saved_model/ |
| D | saved_model_test.py | 955 def assert_num_traces(layer_cls, training_keyword): argument
956 layer = layer_cls()
|