message: Kernel signature reworking, update kernel DepthConcatenation.
diff: @@ -1991,12 +1991,35 @@ void PackWithScaling(const PackParams& params,
  }
  }
  +template <typename Scalar>
  +void DepthConcatenation(const ConcatenationParams& params,
  + const RuntimeShape* const* input_shapes,
  + const Scalar* const* input_data,
  + const RuntimeShape& output_shape, Scalar* output_data) {
  + auto params_copy = params;
  + params_copy.axis = 3;
  + Concatenation(params_copy, input_shapes, input_data, output_shape,
  + output_data);
  +}
  +
  +// TODO(b/80418076): Move to legacy ops file, update invocations.
  template <FusedActivationFunctionType Ac, typename Scalar>
  void DepthConcatenation(const Scalar* const* input_data,
  const Dims<4>* const* input_dims, int inputs_count,
  Scalar* output_data, const Dims<4>& output_dims) {
  - Concatenation<Ac, Scalar>(0, input_data, input_dims, inputs_count,
  - output_data, output_dims);
  + // For now we don't have a model with a Concatenation with fused activation.
  + TFLITE_DCHECK_EQ(Ac, FusedActivationFunctionType::kNone);
  + std::vector<RuntimeShape> input_shapes(inputs_count);
  + std::vector<const RuntimeShape*> input_shapes_indirect(inputs_count);
  + for (int i = 0; i < inputs_count; ++i) {
  + ShapeFromDims(*input_dims[i], &input_shapes[i]);
  + input_shapes_indirect[i] = &input_shapes[i];
  + }
  + tflite::ConcatenationParams op_params;
  + op_params.inputs_count = inputs_count;
  +
  + DepthConcatenation(op_params, input_shapes_indirect.data(), input_data,
  + DimsToShape(output_dims), output_data);
  }
  inline void LstmCell(
  