- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for new_fully_connected (0.29 sec)
-
tensorflow/compiler/mlir/tfr/examples/mnist/mnist_train.py
reshape = tf.reshape(max_pool2, [-1, flatten_size]) # output shape: [-1, 1024] fc1 = gen_mnist_ops.new_fully_connected(reshape, self.weights['f3'], self.biases['b3'], 'RELU') # output shape: [-1, 10] return gen_mnist_ops.new_fully_connected(fc1, self.weights['f4'], self.biases['b4']) def main(strategy):
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 20 03:05:18 UTC 2021 - 6.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/examples/mnist/mnist_ops_test.py
bias = tf.zeros([3]) kwargs = {'input_': input_, 'filter_': filter_, 'bias': bias, 'act': 'RELU'} self._assertOpAndComposite([input_, filter_, bias], tf.function(gen_mnist_ops.new_fully_connected), ops_defs._composite_fully_connected, kwargs) def test_new_max_pool(self): input_ = tf.random.uniform([8, 4, 4, 1]) kwargs = { 'input_': input_,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 28 21:37:05 UTC 2021 - 4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/examples/mnist/ops_defs.py
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 31 20:23:51 UTC 2023 - 6.8K bytes - Viewed (0)