}

// Returns true if the 'op' is one of the supported custom op that takes
// resource type.
bool IsSupportedTFCustomOp(Operation* op) {
  return op->getName().getStringRef().str() == "tf.SentencepieceTokenizeOp";
}

class AnalyzeVariablesPass
    : public impl::AnalyzeVariablesPassBase<AnalyzeVariablesPass> {
 public:
  explicit AnalyzeVariablesPass() {}
  void runOnOperation() override;
};