Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/xla/
H A Dxla_test.py271 self.assertEqual(None, xla.check_function_argument_count(func, 3, None))
273 xla.check_function_argument_count(func, 2, None))
275 self.assertEqual(None, xla.check_function_argument_count(func, 1, queue))
277 xla.check_function_argument_count(func, 2, queue))
285 self.assertEqual(None, xla.check_function_argument_count(func, 3, None))
286 self.assertEqual(None, xla.check_function_argument_count(func, 2, None))
288 xla.check_function_argument_count(func, 1, None))
290 xla.check_function_argument_count(func, 4, None))
292 self.assertEqual(None, xla.check_function_argument_count(func, 2, queue))
293 self.assertEqual(None, xla.check_function_argument_count(func, 1, queue))
[all …]
H A Dxla.py591 def check_function_argument_count(func, input_arity, infeed_queue): function
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dtraining_loop.py67 body_arg_error = xla.check_function_argument_count(
80 condition_arg_error = xla.check_function_argument_count(
H A Dtpu.py1354 arg_error = xla.check_function_argument_count(