.cast<ShapedType>()
               .getElementType()
               .cast<TF::TensorFlowTypeWithSubtype>()
               .GetSubtypes()
               .back();
  }
  return IsSupportedVariableType(type);
}

bool IsSupportedVariableType(ShapedType type) {
  auto element_type = type.getElementType();
  // Check complex types.
  if (auto complex_type = element_type.dyn_cast<mlir::ComplexType>()) {

namespace mlir {
namespace TFL {
namespace utils {

// Returns true if 'op' has type that is supported by native TFLite
// variables.
bool IsSupportedVariableType(Operation* op);

// Returns true if 'type' is supported by native tflite variables.
bool IsSupportedVariableType(ShapedType type);

}  // namespace utils
}  // namespace TFL
}  // namespace mlir

// not.
const char kLegalizeTflVariables[] = "tfl._legalize_tfl_variables";

bool HasSupportedElementType(Operation* op) {
  return utils::IsSupportedVariableType(op);
}

bool IsSupportedElementType(ShapedType type) {
  return utils::IsSupportedVariableType(type);
}

#include "tensorflow/compiler/mlir/lite/transforms/generated_legalize_variables.inc"