Searched refs:hidden_dim (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/wrappers/ |
| D | scikit_learn_test.py | 36 def build_fn_clf(hidden_dim): argument
40 model.add(keras.layers.Dense(hidden_dim))
72 def build_fn_reg(hidden_dim): argument
76 model.add(keras.layers.Dense(hidden_dim))
108 hidden_dim=HIDDEN_DIM,
118 def __call__(self, hidden_dim): argument
119 return build_fn_clf(hidden_dim)
124 hidden_dim=HIDDEN_DIM,
134 def __call__(self, hidden_dim): argument
135 return build_fn_clf(hidden_dim)
[all …]
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/rnn_ptb/ |
| D | rnn_ptb.py | 50 def __init__(self, hidden_dim, num_layers, keep_ratio): argument
54 tf.nn.rnn_cell.BasicLSTMCell(num_units=hidden_dim)
113 hidden_dim, argument
125 num_layers, hidden_dim, dropout=dropout_ratio)
127 self.rnn = RNN(hidden_dim, num_layers, self.keep_ratio)
131 self._output_shape = [-1, hidden_dim]
270 hidden_dim=200,
281 hidden_dim=650,
292 hidden_dim=20,
315 FLAGS.hidden_dim, FLAGS.num_layers, FLAGS.dropout,
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | multi_gpu_utils_test.py | 44 hidden_dim = 10
53 model.add(keras.layers.Dense(hidden_dim,
74 hidden_dim = 10
84 a = keras.layers.Dense(hidden_dim)(input_a)
85 b = keras.layers.Dense(hidden_dim)(input_b)
|