Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/
Dmnist_train.py49 flatten_size = num_features // 16 * n_hidden_2 variable
71 tf.random.truncated_normal([n_hidden_3, flatten_size],
117 reshape = tf.reshape(max_pool2, [-1, flatten_size])