Function Concat

tensorflow/lite/delegates/gpu/metal/kernels/concat.cc:349–359

}

std::vector<ComputeTaskDescriptorPtr> Concat(
    int id, std::vector<ValueId> input_ids, ValueId output_id,
    const ConcatAttributes& attr, const std::vector<BHWC>& input_shapes) {
  if (attr.axis == Axis::CHANNELS) {
    return ConcatZ(id, input_ids, output_id, attr, input_shapes);
  } else if (attr.axis == Axis::WIDTH) {
    return ConcatX(id, input_ids, output_id, attr, input_shapes);
  } else {
    return ConcatY(id, input_ids, output_id, attr, input_shapes);
  }
}

}  // namespace metal
}  // namespace gpu