Searched refs:num_input_edges (Results 1 – 2 of 2) sorted by relevance
789 const int num_input_edges = batch->task(0).inputs.size(); in ProcessBatch() local797 for (int i = 0; i < num_input_edges; ++i) { in ProcessBatch()818 task.context->allocate_output(num_input_edges, index_shape, &output), in ProcessBatch()826 task.context->allocate_output(num_input_edges + 1, in ProcessBatch()833 EmitIndexTensor(last_task_context, *batch, num_input_edges), in ProcessBatch()
1042 int32_t num_input_edges = 0; in Partition() local1057 ++num_input_edges; in Partition()1061 if (num_input_edges != dst->num_inputs()) { in Partition()1063 (dst->num_inputs() - num_input_edges), in Partition()