def setUp(self):
     self._vars = get_test_variables()
     self._n_samples = 1
     self._current_dir = os.path.dirname(os.path.abspath(__file__))
     self._expected_values = joblib.load(
         os.path.join(self._current_dir, 'expected_values.joblib'))
Beispiel #2
0
 def setUp(self):
     self._vars = get_test_variables()
     self._output_tensor = [
         tf.constant([0, 1, 0], dtype=tf.int32),
         tf.constant([1, self._vars['output_dim'], 1], dtype=tf.int32)
     ]
Beispiel #3
0
 def setUp(self):
     self._vars = get_test_variables()
Beispiel #4
0
 def setUp(self):
     self._vars = get_test_variables()
     self._cond_tensor = [
         tf.constant([0, 2, 0, 2], dtype=tf.int32),
         tf.constant([2, 4, 2, 4], dtype=tf.int32)
     ]