Searched refs:input_arity (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | training_loop.py | 69 input_arity = len(inputs)
72 body, input_arity, infeed_queue)
78 input_arity, str([i.name for i in inputs]), body_arg_error))
83 "infeed, but the computation needs %s" % (input_arity, str(
87 condition, input_arity, None)
93 "condition needs %s" % (input_arity, str([i.name for i in inputs]),
100 "condition." % (input_arity, str([i.name for i in inputs]),
105 if input_arity == 0:
114 if input_arity == 0:
183 if input_arity == 0:
|
| D | tpu.py | 1301 input_arity = len(inputs[0])
1304 if len(inputs[i]) != input_arity:
1307 "inputs.".format(input_arity, i, len(inputs[i])))
1316 computation, input_arity, infeed_queue)
1322 input_arity, str([i.name for i in inputs[0]]), arg_error))
1327 " but the computation needs %s" % (input_arity, str(
|
| /external/tensorflow/tensorflow/python/compiler/xla/ |
| D | xla.py | 596 def check_function_argument_count(func, input_arity, infeed_queue): argument
614 num_args_supplied = input_arity
|