Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/kernel_tests/
Dfunctional_ops_test.py853 def _tfMLP(self, xval, wsval, bsval, rewrite_with_while): argument
865 wsval.shape[0],
866 1, [xval, wsval, bsval],
872 def _npMLP(self, xval, wsval, bsval): argument
873 for i in range(wsval.shape[0]):
874 xval = np.tanh(np.dot(xval, wsval[i, :]) + bsval[i, :])
883 wsval = np.random.normal(size=(5, 3, 3))
885 np_ans = self._npMLP(xval, wsval, bsval)
886 tf_for_ans = self._tfMLP(xval, wsval, bsval, rewrite_with_while)