Searched refs:expected_num_outputs (Results 1 – 6 of 6) sorted by relevance
161 Operation* call_op, int expected_num_outputs, in CheckOutputConsumer() argument164 if (num_results != expected_num_outputs) return failure(); in CheckOutputConsumer()166 for (int i = 0; i < expected_num_outputs; ++i) { in CheckOutputConsumer()
78 output_list.expected_num_outputs = 0; in RunModel()83 output_list.expected_num_outputs += 1; in RunModel()
179 unwrap(o)->expected_num_outputs = num_outputs; in TF_OutputListSetNumOutputs()229 int num_outputs = unwrap(o)->expected_num_outputs; in TF_ExecuteOperation()
38 int expected_num_outputs = -1; member
264 output_list.expected_num_outputs = 0; in RunModel()269 output_list.expected_num_outputs += 1; in RunModel()
606 int expected_num_outputs, in CheckNumInputsAndOutputs() argument614 if (node->outputs->size != expected_num_outputs) { in CheckNumInputsAndOutputs()617 node->outputs->size, expected_num_outputs, node_index); in CheckNumInputsAndOutputs()