Searched refs:num_input_edges (Results 1 – 2 of 2) sorted by relevance
652 const int num_input_edges = batch->task(0).inputs.size(); in ProcessBatch() local659 for (int i = 0; i < num_input_edges; ++i) { in ProcessBatch()680 task.context->allocate_output(num_input_edges, index_shape, &output), in ProcessBatch()688 task.context->allocate_output(num_input_edges + 1, in ProcessBatch()695 EmitIndexTensor(last_task_context, *batch, num_input_edges), in ProcessBatch()
1041 int32 num_input_edges = 0; in Partition() local1056 ++num_input_edges; in Partition()1060 if (num_input_edges != dst->num_inputs()) { in Partition()1062 (dst->num_inputs() - num_input_edges), in Partition()