_stan_builtins.py 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558
  1. # -*- coding: utf-8 -*-
  2. """
  3. pygments.lexers._stan_builtins
  4. ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  5. This file contains the names of functions for Stan used by
  6. ``pygments.lexers.math.StanLexer. This is for Stan language version 2.17.0.
  7. :copyright: Copyright 2006-2019 by the Pygments team, see AUTHORS.
  8. :license: BSD, see LICENSE for details.
  9. """
  10. KEYWORDS = (
  11. 'break',
  12. 'continue',
  13. 'else',
  14. 'for',
  15. 'if',
  16. 'in',
  17. 'print',
  18. 'reject',
  19. 'return',
  20. 'while',
  21. )
  22. TYPES = (
  23. 'cholesky_factor_corr',
  24. 'cholesky_factor_cov',
  25. 'corr_matrix',
  26. 'cov_matrix',
  27. 'int',
  28. 'matrix',
  29. 'ordered',
  30. 'positive_ordered',
  31. 'real',
  32. 'row_vector',
  33. 'simplex',
  34. 'unit_vector',
  35. 'vector',
  36. 'void',
  37. )
  38. FUNCTIONS = (
  39. 'abs',
  40. 'acos',
  41. 'acosh',
  42. 'algebra_solver',
  43. 'append_array',
  44. 'append_col',
  45. 'append_row',
  46. 'asin',
  47. 'asinh',
  48. 'atan',
  49. 'atan2',
  50. 'atanh',
  51. 'bernoulli_cdf',
  52. 'bernoulli_lccdf',
  53. 'bernoulli_lcdf',
  54. 'bernoulli_logit_lpmf',
  55. 'bernoulli_logit_rng',
  56. 'bernoulli_lpmf',
  57. 'bernoulli_rng',
  58. 'bessel_first_kind',
  59. 'bessel_second_kind',
  60. 'beta_binomial_cdf',
  61. 'beta_binomial_lccdf',
  62. 'beta_binomial_lcdf',
  63. 'beta_binomial_lpmf',
  64. 'beta_binomial_rng',
  65. 'beta_cdf',
  66. 'beta_lccdf',
  67. 'beta_lcdf',
  68. 'beta_lpdf',
  69. 'beta_rng',
  70. 'binary_log_loss',
  71. 'binomial_cdf',
  72. 'binomial_coefficient_log',
  73. 'binomial_lccdf',
  74. 'binomial_lcdf',
  75. 'binomial_logit_lpmf',
  76. 'binomial_lpmf',
  77. 'binomial_rng',
  78. 'block',
  79. 'categorical_logit_lpmf',
  80. 'categorical_logit_rng',
  81. 'categorical_lpmf',
  82. 'categorical_rng',
  83. 'cauchy_cdf',
  84. 'cauchy_lccdf',
  85. 'cauchy_lcdf',
  86. 'cauchy_lpdf',
  87. 'cauchy_rng',
  88. 'cbrt',
  89. 'ceil',
  90. 'chi_square_cdf',
  91. 'chi_square_lccdf',
  92. 'chi_square_lcdf',
  93. 'chi_square_lpdf',
  94. 'chi_square_rng',
  95. 'cholesky_decompose',
  96. 'choose',
  97. 'col',
  98. 'cols',
  99. 'columns_dot_product',
  100. 'columns_dot_self',
  101. 'cos',
  102. 'cosh',
  103. 'cov_exp_quad',
  104. 'crossprod',
  105. 'csr_extract_u',
  106. 'csr_extract_v',
  107. 'csr_extract_w',
  108. 'csr_matrix_times_vector',
  109. 'csr_to_dense_matrix',
  110. 'cumulative_sum',
  111. 'determinant',
  112. 'diag_matrix',
  113. 'diag_post_multiply',
  114. 'diag_pre_multiply',
  115. 'diagonal',
  116. 'digamma',
  117. 'dims',
  118. 'dirichlet_lpdf',
  119. 'dirichlet_rng',
  120. 'distance',
  121. 'dot_product',
  122. 'dot_self',
  123. 'double_exponential_cdf',
  124. 'double_exponential_lccdf',
  125. 'double_exponential_lcdf',
  126. 'double_exponential_lpdf',
  127. 'double_exponential_rng',
  128. 'e',
  129. 'eigenvalues_sym',
  130. 'eigenvectors_sym',
  131. 'erf',
  132. 'erfc',
  133. 'exp',
  134. 'exp2',
  135. 'exp_mod_normal_cdf',
  136. 'exp_mod_normal_lccdf',
  137. 'exp_mod_normal_lcdf',
  138. 'exp_mod_normal_lpdf',
  139. 'exp_mod_normal_rng',
  140. 'expm1',
  141. 'exponential_cdf',
  142. 'exponential_lccdf',
  143. 'exponential_lcdf',
  144. 'exponential_lpdf',
  145. 'exponential_rng',
  146. 'fabs',
  147. 'falling_factorial',
  148. 'fdim',
  149. 'floor',
  150. 'fma',
  151. 'fmax',
  152. 'fmin',
  153. 'fmod',
  154. 'frechet_cdf',
  155. 'frechet_lccdf',
  156. 'frechet_lcdf',
  157. 'frechet_lpdf',
  158. 'frechet_rng',
  159. 'gamma_cdf',
  160. 'gamma_lccdf',
  161. 'gamma_lcdf',
  162. 'gamma_lpdf',
  163. 'gamma_p',
  164. 'gamma_q',
  165. 'gamma_rng',
  166. 'gaussian_dlm_obs_lpdf',
  167. 'get_lp',
  168. 'gumbel_cdf',
  169. 'gumbel_lccdf',
  170. 'gumbel_lcdf',
  171. 'gumbel_lpdf',
  172. 'gumbel_rng',
  173. 'head',
  174. 'hypergeometric_lpmf',
  175. 'hypergeometric_rng',
  176. 'hypot',
  177. 'inc_beta',
  178. 'int_step',
  179. 'integrate_ode',
  180. 'integrate_ode_bdf',
  181. 'integrate_ode_rk45',
  182. 'inv',
  183. 'inv_chi_square_cdf',
  184. 'inv_chi_square_lccdf',
  185. 'inv_chi_square_lcdf',
  186. 'inv_chi_square_lpdf',
  187. 'inv_chi_square_rng',
  188. 'inv_cloglog',
  189. 'inv_gamma_cdf',
  190. 'inv_gamma_lccdf',
  191. 'inv_gamma_lcdf',
  192. 'inv_gamma_lpdf',
  193. 'inv_gamma_rng',
  194. 'inv_logit',
  195. 'inv_Phi',
  196. 'inv_sqrt',
  197. 'inv_square',
  198. 'inv_wishart_lpdf',
  199. 'inv_wishart_rng',
  200. 'inverse',
  201. 'inverse_spd',
  202. 'is_inf',
  203. 'is_nan',
  204. 'lbeta',
  205. 'lchoose',
  206. 'lgamma',
  207. 'lkj_corr_cholesky_lpdf',
  208. 'lkj_corr_cholesky_rng',
  209. 'lkj_corr_lpdf',
  210. 'lkj_corr_rng',
  211. 'lmgamma',
  212. 'lmultiply',
  213. 'log',
  214. 'log10',
  215. 'log1m',
  216. 'log1m_exp',
  217. 'log1m_inv_logit',
  218. 'log1p',
  219. 'log1p_exp',
  220. 'log2',
  221. 'log_determinant',
  222. 'log_diff_exp',
  223. 'log_falling_factorial',
  224. 'log_inv_logit',
  225. 'log_mix',
  226. 'log_rising_factorial',
  227. 'log_softmax',
  228. 'log_sum_exp',
  229. 'logistic_cdf',
  230. 'logistic_lccdf',
  231. 'logistic_lcdf',
  232. 'logistic_lpdf',
  233. 'logistic_rng',
  234. 'logit',
  235. 'lognormal_cdf',
  236. 'lognormal_lccdf',
  237. 'lognormal_lcdf',
  238. 'lognormal_lpdf',
  239. 'lognormal_rng',
  240. 'machine_precision',
  241. 'matrix_exp',
  242. 'max',
  243. 'mdivide_left_spd',
  244. 'mdivide_left_tri_low',
  245. 'mdivide_right_spd',
  246. 'mdivide_right_tri_low',
  247. 'mean',
  248. 'min',
  249. 'modified_bessel_first_kind',
  250. 'modified_bessel_second_kind',
  251. 'multi_gp_cholesky_lpdf',
  252. 'multi_gp_lpdf',
  253. 'multi_normal_cholesky_lpdf',
  254. 'multi_normal_cholesky_rng',
  255. 'multi_normal_lpdf',
  256. 'multi_normal_prec_lpdf',
  257. 'multi_normal_rng',
  258. 'multi_student_t_lpdf',
  259. 'multi_student_t_rng',
  260. 'multinomial_lpmf',
  261. 'multinomial_rng',
  262. 'multiply_log',
  263. 'multiply_lower_tri_self_transpose',
  264. 'neg_binomial_2_cdf',
  265. 'neg_binomial_2_lccdf',
  266. 'neg_binomial_2_lcdf',
  267. 'neg_binomial_2_log_lpmf',
  268. 'neg_binomial_2_log_rng',
  269. 'neg_binomial_2_lpmf',
  270. 'neg_binomial_2_rng',
  271. 'neg_binomial_cdf',
  272. 'neg_binomial_lccdf',
  273. 'neg_binomial_lcdf',
  274. 'neg_binomial_lpmf',
  275. 'neg_binomial_rng',
  276. 'negative_infinity',
  277. 'normal_cdf',
  278. 'normal_lccdf',
  279. 'normal_lcdf',
  280. 'normal_lpdf',
  281. 'normal_rng',
  282. 'not_a_number',
  283. 'num_elements',
  284. 'ordered_logistic_lpmf',
  285. 'ordered_logistic_rng',
  286. 'owens_t',
  287. 'pareto_cdf',
  288. 'pareto_lccdf',
  289. 'pareto_lcdf',
  290. 'pareto_lpdf',
  291. 'pareto_rng',
  292. 'pareto_type_2_cdf',
  293. 'pareto_type_2_lccdf',
  294. 'pareto_type_2_lcdf',
  295. 'pareto_type_2_lpdf',
  296. 'pareto_type_2_rng',
  297. 'Phi',
  298. 'Phi_approx',
  299. 'pi',
  300. 'poisson_cdf',
  301. 'poisson_lccdf',
  302. 'poisson_lcdf',
  303. 'poisson_log_lpmf',
  304. 'poisson_log_rng',
  305. 'poisson_lpmf',
  306. 'poisson_rng',
  307. 'positive_infinity',
  308. 'pow',
  309. 'print',
  310. 'prod',
  311. 'qr_Q',
  312. 'qr_R',
  313. 'quad_form',
  314. 'quad_form_diag',
  315. 'quad_form_sym',
  316. 'rank',
  317. 'rayleigh_cdf',
  318. 'rayleigh_lccdf',
  319. 'rayleigh_lcdf',
  320. 'rayleigh_lpdf',
  321. 'rayleigh_rng',
  322. 'reject',
  323. 'rep_array',
  324. 'rep_matrix',
  325. 'rep_row_vector',
  326. 'rep_vector',
  327. 'rising_factorial',
  328. 'round',
  329. 'row',
  330. 'rows',
  331. 'rows_dot_product',
  332. 'rows_dot_self',
  333. 'scaled_inv_chi_square_cdf',
  334. 'scaled_inv_chi_square_lccdf',
  335. 'scaled_inv_chi_square_lcdf',
  336. 'scaled_inv_chi_square_lpdf',
  337. 'scaled_inv_chi_square_rng',
  338. 'sd',
  339. 'segment',
  340. 'sin',
  341. 'singular_values',
  342. 'sinh',
  343. 'size',
  344. 'skew_normal_cdf',
  345. 'skew_normal_lccdf',
  346. 'skew_normal_lcdf',
  347. 'skew_normal_lpdf',
  348. 'skew_normal_rng',
  349. 'softmax',
  350. 'sort_asc',
  351. 'sort_desc',
  352. 'sort_indices_asc',
  353. 'sort_indices_desc',
  354. 'sqrt',
  355. 'sqrt2',
  356. 'square',
  357. 'squared_distance',
  358. 'step',
  359. 'student_t_cdf',
  360. 'student_t_lccdf',
  361. 'student_t_lcdf',
  362. 'student_t_lpdf',
  363. 'student_t_rng',
  364. 'sub_col',
  365. 'sub_row',
  366. 'sum',
  367. 'tail',
  368. 'tan',
  369. 'tanh',
  370. 'target',
  371. 'tcrossprod',
  372. 'tgamma',
  373. 'to_array_1d',
  374. 'to_array_2d',
  375. 'to_matrix',
  376. 'to_row_vector',
  377. 'to_vector',
  378. 'trace',
  379. 'trace_gen_quad_form',
  380. 'trace_quad_form',
  381. 'trigamma',
  382. 'trunc',
  383. 'uniform_cdf',
  384. 'uniform_lccdf',
  385. 'uniform_lcdf',
  386. 'uniform_lpdf',
  387. 'uniform_rng',
  388. 'variance',
  389. 'von_mises_lpdf',
  390. 'von_mises_rng',
  391. 'weibull_cdf',
  392. 'weibull_lccdf',
  393. 'weibull_lcdf',
  394. 'weibull_lpdf',
  395. 'weibull_rng',
  396. 'wiener_lpdf',
  397. 'wishart_lpdf',
  398. 'wishart_rng',
  399. )
  400. DISTRIBUTIONS = (
  401. 'bernoulli',
  402. 'bernoulli_logit',
  403. 'beta',
  404. 'beta_binomial',
  405. 'binomial',
  406. 'binomial_logit',
  407. 'categorical',
  408. 'categorical_logit',
  409. 'cauchy',
  410. 'chi_square',
  411. 'dirichlet',
  412. 'double_exponential',
  413. 'exp_mod_normal',
  414. 'exponential',
  415. 'frechet',
  416. 'gamma',
  417. 'gaussian_dlm_obs',
  418. 'gumbel',
  419. 'hypergeometric',
  420. 'inv_chi_square',
  421. 'inv_gamma',
  422. 'inv_wishart',
  423. 'lkj_corr',
  424. 'lkj_corr_cholesky',
  425. 'logistic',
  426. 'lognormal',
  427. 'multi_gp',
  428. 'multi_gp_cholesky',
  429. 'multi_normal',
  430. 'multi_normal_cholesky',
  431. 'multi_normal_prec',
  432. 'multi_student_t',
  433. 'multinomial',
  434. 'neg_binomial',
  435. 'neg_binomial_2',
  436. 'neg_binomial_2_log',
  437. 'normal',
  438. 'ordered_logistic',
  439. 'pareto',
  440. 'pareto_type_2',
  441. 'poisson',
  442. 'poisson_log',
  443. 'rayleigh',
  444. 'scaled_inv_chi_square',
  445. 'skew_normal',
  446. 'student_t',
  447. 'uniform',
  448. 'von_mises',
  449. 'weibull',
  450. 'wiener',
  451. 'wishart',
  452. )
  453. RESERVED = (
  454. 'alignas',
  455. 'alignof',
  456. 'and',
  457. 'and_eq',
  458. 'asm',
  459. 'auto',
  460. 'bitand',
  461. 'bitor',
  462. 'bool',
  463. 'break',
  464. 'case',
  465. 'catch',
  466. 'char',
  467. 'char16_t',
  468. 'char32_t',
  469. 'class',
  470. 'compl',
  471. 'const',
  472. 'const_cast',
  473. 'constexpr',
  474. 'continue',
  475. 'decltype',
  476. 'default',
  477. 'delete',
  478. 'do',
  479. 'double',
  480. 'dynamic_cast',
  481. 'else',
  482. 'enum',
  483. 'explicit',
  484. 'export',
  485. 'extern',
  486. 'false',
  487. 'float',
  488. 'for',
  489. 'friend',
  490. 'fvar',
  491. 'goto',
  492. 'if',
  493. 'in',
  494. 'inline',
  495. 'int',
  496. 'long',
  497. 'lp__',
  498. 'mutable',
  499. 'namespace',
  500. 'new',
  501. 'noexcept',
  502. 'not',
  503. 'not_eq',
  504. 'nullptr',
  505. 'operator',
  506. 'or',
  507. 'or_eq',
  508. 'private',
  509. 'protected',
  510. 'public',
  511. 'register',
  512. 'reinterpret_cast',
  513. 'repeat',
  514. 'return',
  515. 'short',
  516. 'signed',
  517. 'sizeof',
  518. 'STAN_MAJOR',
  519. 'STAN_MATH_MAJOR',
  520. 'STAN_MATH_MINOR',
  521. 'STAN_MATH_PATCH',
  522. 'STAN_MINOR',
  523. 'STAN_PATCH',
  524. 'static',
  525. 'static_assert',
  526. 'static_cast',
  527. 'struct',
  528. 'switch',
  529. 'template',
  530. 'then',
  531. 'this',
  532. 'thread_local',
  533. 'throw',
  534. 'true',
  535. 'try',
  536. 'typedef',
  537. 'typeid',
  538. 'typename',
  539. 'union',
  540. 'unsigned',
  541. 'until',
  542. 'using',
  543. 'var',
  544. 'virtual',
  545. 'void',
  546. 'volatile',
  547. 'wchar_t',
  548. 'while',
  549. 'xor',
  550. 'xor_eq',
  551. )