Home
last modified time | relevance | path

Searched refs:noise_shape (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/contrib/nn/python/ops/
Dalpha_dropout_test.py51 _ = alpha_dropout(t, keep_prob, noise_shape=[x_dim, y_dim + 10])
53 _ = alpha_dropout(t, keep_prob, noise_shape=[x_dim, y_dim, 5])
55 _ = alpha_dropout(t, keep_prob, noise_shape=[x_dim + 3])
57 _ = alpha_dropout(t, keep_prob, noise_shape=[x_dim])
60 _ = alpha_dropout(t, keep_prob, noise_shape=[y_dim])
61 _ = alpha_dropout(t, keep_prob, noise_shape=[1, y_dim])
62 _ = alpha_dropout(t, keep_prob, noise_shape=[x_dim, 1])
63 _ = alpha_dropout(t, keep_prob, noise_shape=[1, 1])
Dalpha_dropout.py30 def alpha_dropout(x, keep_prob, noise_shape=None, seed=None, name=None): # pylint: disable=invalid-… argument
72 noise_shape = noise_shape if noise_shape is not None else array_ops.shape(x)
73 random_tensor = random_ops.random_uniform(noise_shape,
/external/tensorflow/tensorflow/python/keras/layers/
Dnoise.py163 def __init__(self, rate, noise_shape=None, seed=None, **kwargs): argument
166 self.noise_shape = noise_shape
171 return self.noise_shape if self.noise_shape else array_ops.shape(inputs)
175 noise_shape = self._get_noise_shape(inputs)
183 K.random_uniform(noise_shape, seed=seed), rate)
Dcore.py133 def __init__(self, rate, noise_shape=None, seed=None, **kwargs): argument
136 self.noise_shape = noise_shape
144 if self.noise_shape is None:
145 return self.noise_shape
146 return nn_ops._get_noise_shape(inputs, self.noise_shape) # pylint: disable=protected-access
155 noise_shape=self._get_noise_shape(inputs),
170 'noise_shape': self.noise_shape,
215 noise_shape = (input_shape[0], 1, input_shape[2])
216 return noise_shape
/external/tensorflow/tensorflow/python/layers/
Dcore.py216 noise_shape=None, argument
221 noise_shape=noise_shape,
236 noise_shape=None, argument
271 layer = Dropout(rate, noise_shape=noise_shape, seed=seed, name=name)
Dcore_test.py389 self.assertEqual(dp.noise_shape, None)
422 noise_shape = [None, 1, None]
423 dp = core_layers.Dropout(0.5, noise_shape=noise_shape, seed=1)
432 noise_shape = [5, 1, 2]
433 dp = core_layers.Dropout(0.5, noise_shape=noise_shape, seed=1)
/external/tensorflow/tensorflow/python/ops/
Dnn_test.py343 dropout = nn_ops.dropout(t, keep_prob, noise_shape=[x_dim, 1])
367 dropout = nn_ops.dropout(t, keep_prob, noise_shape=[x_dim, 1])
412 x, keep_prob, noise_shape=array_ops.placeholder(dtypes.int32))
422 dropout = nn_ops.dropout(t, keep_prob, noise_shape=[None, 1])
475 _ = nn_ops.dropout(t, keep_prob, noise_shape=[x_dim, y_dim + 10])
477 _ = nn_ops.dropout(t, keep_prob, noise_shape=[x_dim, y_dim, 5])
479 _ = nn_ops.dropout(t, keep_prob, noise_shape=[x_dim + 3])
481 _ = nn_ops.dropout(t, keep_prob, noise_shape=[x_dim])
483 _ = nn_ops.dropout(t, keep_prob, noise_shape=[y_dim])
484 _ = nn_ops.dropout(t, keep_prob, noise_shape=[1, y_dim])
[all …]
Dnn_ops.py4017 def _get_noise_shape(x, noise_shape): argument
4019 if noise_shape is None:
4026 noise_shape_ = tensor_shape.as_shape(noise_shape)
4028 return noise_shape
4039 return noise_shape
4046 def dropout(x, keep_prob=None, noise_shape=None, seed=None, name=None, argument
4093 return dropout_v2(x, rate, noise_shape=noise_shape, seed=seed, name=name)
4097 def dropout_v2(x, rate, noise_shape=None, seed=None, name=None): argument
4156 noise_shape = _get_noise_shape(x, noise_shape)
4163 noise_shape, seed=seed, dtype=x.dtype)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.keras.layers.-alpha-dropout.pbtxt89 …argspec: "args=[\'self\', \'rate\', \'noise_shape\', \'seed\'], varargs=None, keywords=kwargs, def…
Dtensorflow.keras.layers.-dropout.pbtxt89 …argspec: "args=[\'self\', \'rate\', \'noise_shape\', \'seed\'], varargs=None, keywords=kwargs, def…
Dtensorflow.layers.-dropout.pbtxt99 …argspec: "args=[\'self\', \'rate\', \'noise_shape\', \'seed\', \'name\'], varargs=None, keywords=k…
Dtensorflow.layers.pbtxt125 …argspec: "args=[\'inputs\', \'rate\', \'noise_shape\', \'seed\', \'training\', \'name\'], varargs=…
Dtensorflow.nn.pbtxt177 …argspec: "args=[\'x\', \'keep_prob\', \'noise_shape\', \'seed\', \'name\', \'rate\'], varargs=None…
Dtensorflow.keras.backend.pbtxt141 …argspec: "args=[\'x\', \'level\', \'noise_shape\', \'seed\'], varargs=None, keywords=None, default…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.keras.layers.-alpha-dropout.pbtxt89 …argspec: "args=[\'self\', \'rate\', \'noise_shape\', \'seed\'], varargs=None, keywords=kwargs, def…
Dtensorflow.keras.layers.-dropout.pbtxt89 …argspec: "args=[\'self\', \'rate\', \'noise_shape\', \'seed\'], varargs=None, keywords=kwargs, def…
Dtensorflow.nn.pbtxt141 …argspec: "args=[\'x\', \'rate\', \'noise_shape\', \'seed\', \'name\'], varargs=None, keywords=None…
Dtensorflow.keras.backend.pbtxt141 …argspec: "args=[\'x\', \'level\', \'noise_shape\', \'seed\'], varargs=None, keywords=None, default…
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dlayers.py1565 noise_shape=None, argument
1598 noise_shape=noise_shape,
/external/tensorflow/tensorflow/python/keras/
Dbackend.py4069 def dropout(x, level, noise_shape=None, seed=None): argument
4085 return nn.dropout_v2(x, rate=level, noise_shape=noise_shape, seed=seed)
Dbackend_test.py1366 outputs = keras.backend.dropout(inputs, 0.2, noise_shape=(200, 1))