Searched refs:num_input_edges (Results 1 – 2 of 2) sorted by relevance
570 const int num_input_edges = batch->task(0).inputs.size(); in ProcessBatch() local577 for (int i = 0; i < num_input_edges; ++i) { in ProcessBatch()599 task.context->allocate_output(num_input_edges, index_shape, &output), in ProcessBatch()607 task.context->allocate_output(num_input_edges + 1, in ProcessBatch()614 EmitIndexTensor(last_task_context, *batch, num_input_edges), in ProcessBatch()
1014 int32 num_input_edges = 0; in Partition() local1029 ++num_input_edges; in Partition()1033 if (num_input_edges != dst->num_inputs()) { in Partition()1035 (dst->num_inputs() - num_input_edges), in Partition()