Searched refs:input_arity (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | training_loop.py | 61 input_arity = len(inputs)
64 body, input_arity, infeed_queue)
70 input_arity, str([i.name for i in inputs]), body_arg_error))
75 "infeed, but the computation needs %s" % (input_arity, str(
79 condition, input_arity, None)
85 "condition needs %s" % (input_arity, str([i.name for i in inputs]),
92 "condition." % (input_arity, str([i.name for i in inputs]),
97 if input_arity == 0:
106 if input_arity == 0:
175 if input_arity == 0:
|
| D | xla.py | 61 def check_function_argument_count(func, input_arity, infeed_queue): argument
79 num_args_supplied = input_arity
|
| D | tpu.py | 764 input_arity = len(inputs[0])
767 if len(inputs[i]) != input_arity:
770 "inputs.".format(input_arity, i, len(inputs[i])))
779 computation, input_arity, infeed_queue)
785 input_arity, str([i.name for i in inputs[0]]), arg_error))
790 " but the computation needs %s" % (input_arity, str(
|
| /external/tensorflow/tensorflow/contrib/compiler/ |
| D | xla.py | 793 def check_function_argument_count(func, input_arity, infeed_queue): argument
811 num_args_supplied = input_arity
|