| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | noise.py | 169 def __init__(self, rate, noise_shape=None, seed=None, **kwargs): argument
172 self.noise_shape = noise_shape
177 return self.noise_shape if self.noise_shape else array_ops.shape(inputs)
181 noise_shape = self._get_noise_shape(inputs)
189 K.random_uniform(noise_shape, seed=seed), rate)
|
| D | core.py | 194 def __init__(self, rate, noise_shape=None, seed=None, **kwargs): argument
201 self.noise_shape = noise_shape
209 if self.noise_shape is None:
213 noise_shape = []
214 for i, value in enumerate(self.noise_shape):
215 noise_shape.append(concrete_inputs_shape[i] if value is None else value)
216 return ops.convert_to_tensor_v2_with_dispatch(noise_shape)
225 noise_shape=self._get_noise_shape(inputs),
239 'noise_shape': self.noise_shape,
284 noise_shape = (input_shape[0], 1, input_shape[2])
[all …]
|
| D | core_test.py | 88 layer = keras.layers.Dropout(0.5, noise_shape=(None, 1, None))
|
| /external/tensorflow/tensorflow/python/keras/legacy_tf_layers/ |
| D | core.py | 216 noise_shape=None, argument
221 noise_shape=noise_shape,
233 noise_shape=None, argument
271 layer = Dropout(rate, noise_shape=noise_shape, seed=seed, name=name)
|
| D | core_test.py | 374 self.assertEqual(dp.noise_shape, None)
407 noise_shape = [None, 1, None]
408 dp = core_layers.Dropout(0.5, noise_shape=noise_shape, seed=1)
417 noise_shape = [5, 1, 2]
418 dp = core_layers.Dropout(0.5, noise_shape=noise_shape, seed=1)
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_dispatch.py | 461 def _ragged_nn_dropout_v1(x, keep_prob=None, noise_shape=None, seed=None, argument
463 if noise_shape is not None:
471 def _ragged_nn_dropout_v2(x, rate, noise_shape=None, seed=None, name=None): argument
472 if noise_shape is not None:
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_test.py | 373 dropout = nn_ops.dropout(t, rate=(1 - keep_prob), noise_shape=[x_dim, 1])
397 dropout = nn_ops.dropout(t, rate=(1 - keep_prob), noise_shape=[x_dim, 1])
444 noise_shape=array_ops.placeholder(dtypes.int32))
454 dropout = nn_ops.dropout(t, rate=(1 - keep_prob), noise_shape=[None, 1])
518 t, rate=(1 - keep_prob), noise_shape=[x_dim, y_dim + 10])
520 _ = nn_ops.dropout(t, rate=(1 - keep_prob), noise_shape=[x_dim, y_dim, 5])
522 _ = nn_ops.dropout(t, rate=(1 - keep_prob), noise_shape=[x_dim + 3])
524 _ = nn_ops.dropout(t, rate=(1 - keep_prob), noise_shape=[x_dim])
526 _ = nn_ops.dropout(t, rate=(1 - keep_prob), noise_shape=[y_dim])
527 _ = nn_ops.dropout(t, rate=(1 - keep_prob), noise_shape=[1, y_dim])
[all …]
|
| D | nn_ops.py | 4981 def _get_noise_shape(x, noise_shape): argument
4983 if noise_shape is None:
4990 noise_shape_ = tensor_shape.as_shape(noise_shape)
4992 return noise_shape
5003 return noise_shape
5011 def dropout(x, keep_prob=None, noise_shape=None, seed=None, name=None, argument
5058 return dropout_v2(x, rate, noise_shape=noise_shape, seed=seed, name=name)
5063 def dropout_v2(x, rate, noise_shape=None, seed=None, name=None): argument
5179 noise_shape = _get_noise_shape(x, noise_shape)
5186 noise_shape, seed=seed, dtype=x_dtype)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.keras.layers.-alpha-dropout.pbtxt | 132 …argspec: "args=[\'self\', \'rate\', \'noise_shape\', \'seed\'], varargs=None, keywords=kwargs, def…
|
| D | tensorflow.keras.layers.-dropout.pbtxt | 132 …argspec: "args=[\'self\', \'rate\', \'noise_shape\', \'seed\'], varargs=None, keywords=kwargs, def…
|
| D | tensorflow.layers.-dropout.pbtxt | 142 …argspec: "args=[\'self\', \'rate\', \'noise_shape\', \'seed\', \'name\'], varargs=None, keywords=k…
|
| D | tensorflow.layers.pbtxt | 125 …argspec: "args=[\'inputs\', \'rate\', \'noise_shape\', \'seed\', \'training\', \'name\'], varargs=…
|
| D | tensorflow.keras.backend.pbtxt | 145 …argspec: "args=[\'x\', \'level\', \'noise_shape\', \'seed\'], varargs=None, keywords=None, default…
|
| D | tensorflow.nn.pbtxt | 177 …argspec: "args=[\'x\', \'keep_prob\', \'noise_shape\', \'seed\', \'name\', \'rate\'], varargs=None…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.layers.-alpha-dropout.pbtxt | 132 …argspec: "args=[\'self\', \'rate\', \'noise_shape\', \'seed\'], varargs=None, keywords=kwargs, def…
|
| D | tensorflow.keras.layers.-dropout.pbtxt | 132 …argspec: "args=[\'self\', \'rate\', \'noise_shape\', \'seed\'], varargs=None, keywords=kwargs, def…
|
| D | tensorflow.nn.pbtxt | 141 …argspec: "args=[\'x\', \'rate\', \'noise_shape\', \'seed\', \'name\'], varargs=None, keywords=None…
|
| D | tensorflow.keras.backend.pbtxt | 141 …argspec: "args=[\'x\', \'level\', \'noise_shape\', \'seed\'], varargs=None, keywords=None, default…
|
| /external/tensorflow/tensorflow/python/keras/applications/ |
| D | efficientnet.py | 516 drop_rate, noise_shape=(None, 1, 1, 1), name=name + 'drop')(x)
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | benchmarks_test.py | 1090 noise_shape=None, argument
1098 self._m_2_by_2, rate=rate, noise_shape=noise_shape)
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 5185 def dropout(x, level, noise_shape=None, seed=None): argument
5201 return nn.dropout_v2(x, rate=level, noise_shape=noise_shape, seed=seed)
|
| D | backend_test.py | 1613 outputs = backend.dropout(inputs, 0.2, noise_shape=(200, 1))
|