Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/
Dmnist_train.py47 n_hidden_2 = 64 # 2nd conv layer number of neurons. variable
49 flatten_size = num_features // 16 * n_hidden_2
66 tf.random.truncated_normal([5, 5, n_hidden_1, n_hidden_2],
83 'b2': tf.Variable(tf.zeros([n_hidden_2])),