Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/compiler/xla/
Dxla_test.py275 self.assertEqual(None, xla.check_function_argument_count(func, 3, None))
277 xla.check_function_argument_count(func, 2, None))
279 self.assertEqual(None, xla.check_function_argument_count(func, 1, queue))
281 xla.check_function_argument_count(func, 2, queue))
289 self.assertEqual(None, xla.check_function_argument_count(func, 3, None))
290 self.assertEqual(None, xla.check_function_argument_count(func, 2, None))
292 xla.check_function_argument_count(func, 1, None))
294 xla.check_function_argument_count(func, 4, None))
296 self.assertEqual(None, xla.check_function_argument_count(func, 2, queue))
297 self.assertEqual(None, xla.check_function_argument_count(func, 1, queue))
[all …]
Dxla.py596 def check_function_argument_count(func, input_arity, infeed_queue): function
/external/tensorflow/tensorflow/python/tpu/
Dtraining_loop.py71 body_arg_error = xla.check_function_argument_count(
86 condition_arg_error = xla.check_function_argument_count(
Dtpu.py1315 arg_error = xla.check_function_argument_count(