Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/compiler/
Dxla_test.py204 self.assertEqual(None, xla.check_function_argument_count(func, 3, None))
206 xla.check_function_argument_count(func, 2, None))
208 self.assertEqual(None, xla.check_function_argument_count(func, 1, queue))
210 xla.check_function_argument_count(func, 2, queue))
218 self.assertEqual(None, xla.check_function_argument_count(func, 3, None))
219 self.assertEqual(None, xla.check_function_argument_count(func, 2, None))
221 xla.check_function_argument_count(func, 1, None))
223 xla.check_function_argument_count(func, 4, None))
225 self.assertEqual(None, xla.check_function_argument_count(func, 2, queue))
226 self.assertEqual(None, xla.check_function_argument_count(func, 1, queue))
[all …]
Dxla.py793 def check_function_argument_count(func, input_arity, infeed_queue): function
/external/tensorflow/tensorflow/python/tpu/
Dtraining_loop.py63 body_arg_error = xla.check_function_argument_count(
78 condition_arg_error = xla.check_function_argument_count(
Dxla.py61 def check_function_argument_count(func, input_arity, infeed_queue): function
Dtpu.py778 arg_error = xla.check_function_argument_count(