}
  }
  return not_found;
}

// Skip the assertions because they currently create problematic dependencies.
constexpr bool kDoAssertions = true;

void AddAssertion(OpBuilder& builder, Location& loc, Value cond,
                  const std::string& message) {
  if (!kDoAssertions) return;
  auto shape_type =
      RankedTensorType::get({1}, builder.getType<TF::StringType>());