Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/
Dmnist_train.py46 n_hidden_1 = 32 # 1st conv layer number of neurons. variable
61 tf.random.truncated_normal([5, 5, num_channels, n_hidden_1],
66 tf.random.truncated_normal([5, 5, n_hidden_1, n_hidden_2],
82 'b1': tf.Variable(tf.zeros([n_hidden_1])),