hailo-metadata.json 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050
  1. [
  2. {
  3. "name": "input_layer",
  4. "description": "Represents an input of the model",
  5. "attributes": [
  6. {
  7. "name": "original_names",
  8. "type": "string[]",
  9. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  10. "visible": false
  11. }
  12. ]
  13. },
  14. {
  15. "name": "output_layer",
  16. "description": "Represents an output of the model",
  17. "attributes": [
  18. {
  19. "name": "original_names",
  20. "type": "string[]",
  21. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  22. "visible": false
  23. }]
  24. },
  25. {
  26. "name": "postprocess",
  27. "description": "Represents a whole post-processing function of some meta-architecture",
  28. "attributes": [
  29. {
  30. "name": "original_names",
  31. "type": "string[]",
  32. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  33. "visible": false
  34. },
  35. {
  36. "name": "max_proposals_per_class",
  37. "type": "int64",
  38. "description": "Maximum number of proposals per class",
  39. "visible": false
  40. },
  41. {
  42. "name": "iou_th",
  43. "type": "float32",
  44. "visible": false,
  45. "description": "Intersection over union overlap threshold, used in the NMS iterative elimination process where potential duplicates of detected items are ignored"
  46. },
  47. {
  48. "name": "meta_arch",
  49. "type": "string",
  50. "visible": false,
  51. "description": "Postprocessing meta-architecture name"
  52. },
  53. {
  54. "name": "max_total_output_proposals",
  55. "type": "int64",
  56. "visible": false,
  57. "description": "Maximum number of bounding box proposals"
  58. },
  59. {
  60. "name": "postprocess_type",
  61. "type": "string",
  62. "visible": false,
  63. "description": "Postprocessing type name"
  64. }
  65. ]
  66. },
  67. {
  68. "name": "conv",
  69. "category": "Layer",
  70. "description": "Convolution layer",
  71. "attributes": [
  72. {
  73. "name": "original_names",
  74. "type": "string[]",
  75. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  76. "visible": false
  77. },
  78. {
  79. "name": "kernel_shape",
  80. "type": "int64[]",
  81. "label": "kernel",
  82. "description": "Shape of the kernel in Tensorflow convention (kernel height, kernel width, features in, features out)",
  83. "visible": true
  84. },
  85. {
  86. "name": "strides",
  87. "type": "int64[]",
  88. "description": "Stride along each axis (batch, height, width, features)"
  89. },
  90. {
  91. "name": "dilations",
  92. "type": "int64[]",
  93. "description": "Dilation value along each axis (batch, height, width, features)"
  94. },
  95. {
  96. "name": "padding",
  97. "type": "string",
  98. "description": "Padding mode, either VALID, SAME (symmetric, Caffe-like), SAME_TENSORFLOW, or DECONV"
  99. },
  100. {
  101. "name": "groups",
  102. "type": "int64",
  103. "description": "Number of groups input channels and output channels are divided into"
  104. },
  105. {
  106. "name": "batch_norm",
  107. "type": "boolean",
  108. "description": "Whether batch normalization is folded into the layer"
  109. },
  110. {
  111. "name": "elementwise_add",
  112. "type": "boolean",
  113. "description": "Whether elementwise addition is folded into the layer",
  114. "visible": false
  115. },
  116. {
  117. "name": "activation",
  118. "type": "string",
  119. "description": "Activation function name",
  120. "visible": false
  121. },
  122. {
  123. "name": "pre_layer_batch_norm",
  124. "type": "boolean",
  125. "description": "Whether batch normalization is folded into the layer, before the operation itself",
  126. "visible": false
  127. },
  128. {
  129. "name": "transpose_output_width_features",
  130. "type": "boolean",
  131. "description": "Whether to transpose the width and the features axes of the layer's output tensor",
  132. "visible": false
  133. },
  134. {
  135. "name": "spatial_flatten_output",
  136. "type": "boolean",
  137. "description": "Whether to flatten the layer's output to one row",
  138. "visible": false
  139. }
  140. ]
  141. },
  142. {
  143. "name": "relu",
  144. "category": "Activation"
  145. },
  146. {
  147. "name": "delta",
  148. "category": "Activation"
  149. },
  150. {
  151. "name": "activation",
  152. "category": "Activation",
  153. "description": "Activation function",
  154. "attributes": [
  155. {
  156. "name": "original_names",
  157. "type": "string[]",
  158. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  159. "visible": false
  160. },
  161. {
  162. "name": "batch_norm",
  163. "type": "boolean",
  164. "description": "Whether batch normalization is folded into the layer",
  165. "visible": false
  166. },
  167. {
  168. "name": "elementwise_add",
  169. "type": "boolean",
  170. "description": "Whether elementwise addition is folded into the layer",
  171. "visible": false
  172. },
  173. {
  174. "name": "activation",
  175. "type": "string",
  176. "description": "Activation function name",
  177. "visible": false
  178. }
  179. ]
  180. },
  181. {
  182. "name": "argmax",
  183. "description": "Argmax layer",
  184. "attributes": [
  185. {
  186. "name": "original_names",
  187. "type": "string[]",
  188. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  189. "visible": false
  190. }]
  191. },
  192. {
  193. "name": "avgpool",
  194. "category": "Pool",
  195. "description": "Average pooling layer",
  196. "attributes": [
  197. {
  198. "name": "original_names",
  199. "type": "string[]",
  200. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  201. "visible": false
  202. },
  203. {
  204. "name": "kernel_shape",
  205. "type": "int64[]",
  206. "label": "kernel",
  207. "description": "Shape of the kernel in Tensorflow convention (kernel height, kernel width, features in, features out)",
  208. "visible": true
  209. },
  210. {
  211. "name": "strides",
  212. "type": "int64[]",
  213. "description": "Stride along each axis (batch, height, width, features)",
  214. "visible": false
  215. },
  216. {
  217. "name": "padding",
  218. "type": "string",
  219. "description": "Padding mode, either VALID, SAME (symmetric, Caffe-like), SAME_TENSORFLOW, or DECONV",
  220. "visible": false
  221. },
  222. {
  223. "name": "activation",
  224. "type": "string",
  225. "description": "Activation function name",
  226. "visible": false
  227. }
  228. ]
  229. },
  230. {
  231. "name": "batch_norm",
  232. "category": "Normalization",
  233. "description": "Batch normalization layer",
  234. "attributes": [
  235. {
  236. "name": "original_names",
  237. "type": "string[]",
  238. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  239. "visible": false
  240. },
  241. {
  242. "name": "elementwise_add",
  243. "type": "boolean",
  244. "description": "Whether elementwise addition is folded into the layer",
  245. "visible": false
  246. },
  247. {
  248. "name": "activation",
  249. "type": "string",
  250. "description": "Activation function name",
  251. "visible": false
  252. }
  253. ]
  254. },
  255. {
  256. "name": "bbox_decoder",
  257. "description": "Bounding box decoding layer",
  258. "attributes": [
  259. {
  260. "name": "original_names",
  261. "type": "string[]",
  262. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  263. "visible": false
  264. }
  265. ]
  266. },
  267. {
  268. "name": "deconv",
  269. "category": "Layer",
  270. "description": "Deconvolution (transposed convolution) layer",
  271. "attributes": [
  272. {
  273. "name": "original_names",
  274. "type": "string[]",
  275. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  276. "visible": false
  277. },
  278. {
  279. "name": "kernel_shape",
  280. "type": "int64[]",
  281. "label": "kernel",
  282. "description": "Shape of the kernel in Tensorflow convention (kernel height, kernel width, features in, features out)",
  283. "visible": true
  284. },
  285. {
  286. "name": "strides",
  287. "type": "int64[]",
  288. "description": "Stride along each axis (batch, height, width, features)",
  289. "visible": false
  290. },
  291. {
  292. "name": "dilations",
  293. "type": "int64[]",
  294. "description": "Dilation value along each axis (batch, height, width, features)",
  295. "visible": false
  296. },
  297. {
  298. "name": "padding",
  299. "type": "string",
  300. "description": "Padding mode, either VALID, SAME (symmetric, Caffe-like), SAME_TENSORFLOW, or DECONV",
  301. "visible": false
  302. },
  303. {
  304. "name": "groups",
  305. "type": "int64",
  306. "description": "Number of groups input channels and output channels are divided into",
  307. "visible": false
  308. },
  309. {
  310. "name": "batch_norm",
  311. "type": "boolean",
  312. "description": "Whether batch normalization is folded into the layer",
  313. "visible": false
  314. },
  315. {
  316. "name": "elementwise_add",
  317. "type": "boolean",
  318. "description": "Whether elementwise addition is folded into the layer",
  319. "visible": false
  320. },
  321. {
  322. "name": "activation",
  323. "type": "string",
  324. "description": "Activation function name",
  325. "visible": false
  326. }
  327. ]
  328. },
  329. {
  330. "name": "dense",
  331. "category": "Layer",
  332. "description": "Dense (fully connected) layer",
  333. "attributes": [
  334. {
  335. "name": "original_names",
  336. "type": "string[]",
  337. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  338. "visible": false
  339. },
  340. {
  341. "name": "kernel_shape",
  342. "type": "int64[]",
  343. "label": "kernel",
  344. "description": "Shape of the kernel in Tensorflow convention (kernel height, kernel width, features in, features out)",
  345. "visible": true
  346. },
  347. {
  348. "name": "batch_norm",
  349. "type": "boolean",
  350. "description": "Whether batch normalization is folded into the layer",
  351. "visible": false
  352. }
  353. ]
  354. },
  355. {
  356. "name": "depth_to_space",
  357. "description": "Depth to space layer",
  358. "attributes": [
  359. {
  360. "name": "original_names",
  361. "type": "string[]",
  362. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  363. "visible": false
  364. },
  365. {
  366. "name": "block_sizes",
  367. "type": "int64[]",
  368. "description": "Block size along each spatial axis",
  369. "visible": false
  370. },
  371. {
  372. "name": "depth_to_space_type",
  373. "type": "string",
  374. "description": "Depth to space variant, either dcr (depth-column-row) or crd (column-row-depth)",
  375. "visible": false
  376. }
  377. ]
  378. },
  379. {
  380. "name": "dw",
  381. "description": "Depthwise convolution layer",
  382. "attributes": [
  383. {
  384. "name": "original_names",
  385. "type": "string[]",
  386. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  387. "visible": false
  388. },
  389. {
  390. "name": "kernel_shape",
  391. "type": "int64[]",
  392. "label": "kernel",
  393. "description": "Shape of the kernel in Tensorflow convention (kernel height, kernel width, features in, features out)",
  394. "visible": true
  395. },
  396. {
  397. "name": "strides",
  398. "type": "int64[]",
  399. "description": "Stride along each axis (batch, height, width, features)",
  400. "visible": false
  401. },
  402. {
  403. "name": "dilations",
  404. "type": "int64[]",
  405. "description": "Dilation value along each axis (batch, height, width, features)",
  406. "visible": false
  407. },
  408. {
  409. "name": "padding",
  410. "type": "string",
  411. "description": "Padding mode, either VALID, SAME (symmetric, Caffe-like), SAME_TENSORFLOW, or DECONV",
  412. "visible": false
  413. },
  414. {
  415. "name": "groups",
  416. "type": "int64",
  417. "description": "Number of groups input channels and output channels are divided into",
  418. "visible": false
  419. },
  420. {
  421. "name": "batch_norm",
  422. "type": "boolean",
  423. "description": "Whether batch normalization is folded into the layer",
  424. "visible": false
  425. },
  426. {
  427. "name": "elementwise_add",
  428. "type": "boolean",
  429. "description": "Whether elementwise addition is folded into the layer",
  430. "visible": false
  431. },
  432. {
  433. "name": "activation",
  434. "type": "string",
  435. "description": "Activation function name",
  436. "visible": false
  437. },
  438. {
  439. "name": "transpose_output_width_features",
  440. "type": "string",
  441. "description": "Whether to transpose the width and the features axes of the layer's output tensor",
  442. "visible": false
  443. },
  444. {
  445. "name": "dynamic_weights",
  446. "type": "boolean",
  447. "description": "Whether the layer's weights are data driven",
  448. "visible": false
  449. }
  450. ]
  451. },
  452. {
  453. "name": "external_pad",
  454. "description": "Padding layer",
  455. "attributes": [
  456. {
  457. "name": "original_names",
  458. "type": "string[]",
  459. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  460. "visible": false
  461. },
  462. {
  463. "name": "external_pad_params",
  464. "type": "int64[]",
  465. "description": "Padding value in pixels in each edge (top, bottom, left, right)",
  466. "visible": false
  467. }
  468. ]
  469. },
  470. {
  471. "name": "feature_interleave",
  472. "description": "Feature interleave layer",
  473. "attributes": [
  474. {
  475. "name": "original_names",
  476. "type": "string[]",
  477. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  478. "visible": false
  479. }]
  480. },
  481. {
  482. "name": "feature_multiplier",
  483. "description": "Elementwise feature multiplication layer",
  484. "attributes": [
  485. {
  486. "name": "original_names",
  487. "type": "string[]",
  488. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  489. "visible": false
  490. },
  491. {
  492. "name": "feature_multiplier_type",
  493. "type": "string",
  494. "description": "Feature multiplier variant, either square (to multiply each value by itself), or user_specified",
  495. "visible": false
  496. }
  497. ]
  498. },
  499. {
  500. "name": "feature_shuffle",
  501. "description": "Feature shuffle layer",
  502. "attributes": [
  503. {
  504. "name": "original_names",
  505. "type": "string[]",
  506. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  507. "visible": false
  508. }]
  509. },
  510. {
  511. "name": "format_conversion",
  512. "description": "Reshapes the input tensor between different memory layouts",
  513. "attributes": [
  514. {
  515. "name": "original_names",
  516. "type": "string[]",
  517. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  518. "visible": false
  519. },
  520. {
  521. "name": "expand_spatial_sizes",
  522. "type": "int64[]",
  523. "description": "New output tensor dimensions after the reshape (height, width)",
  524. "visible": false
  525. },
  526. {
  527. "name": "conversion_type",
  528. "type": "string",
  529. "visible": false,
  530. "description": "Format conversion variant"
  531. }
  532. ]
  533. },
  534. {
  535. "name": "global_avg_pool",
  536. "category": "Pool",
  537. "description": "Global average pooling layer",
  538. "attributes": [
  539. {
  540. "name": "original_names",
  541. "type": "string[]",
  542. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  543. "visible": false
  544. }]
  545. },
  546. {
  547. "name": "maxpool",
  548. "category": "Pool",
  549. "description": "Maximum pooling layer",
  550. "attributes": [
  551. {
  552. "name": "original_names",
  553. "type": "string[]",
  554. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  555. "visible": false
  556. },
  557. {
  558. "name": "kernel_shape",
  559. "type": "int64[]",
  560. "label": "kernel",
  561. "description": "Shape of the kernel in Tensorflow convention (kernel height, kernel width, features in, features out)",
  562. "visible": true
  563. },
  564. {
  565. "name": "strides",
  566. "type": "int64[]",
  567. "description": "Stride along each axis (batch, height, width, features)",
  568. "visible": false
  569. },
  570. {
  571. "name": "padding",
  572. "type": "string",
  573. "description": "Padding mode, either VALID, SAME (symmetric, Caffe-like), SAME_TENSORFLOW, or DECONV",
  574. "visible": false
  575. },
  576. {
  577. "name": "activation",
  578. "type": "string",
  579. "description": "Activation function name",
  580. "visible": false
  581. }
  582. ]
  583. },
  584. {
  585. "name": "nms",
  586. "description": "Non-maximum suppression layer",
  587. "attributes": [
  588. {
  589. "name": "original_names",
  590. "type": "string[]",
  591. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  592. "visible": false
  593. },
  594. {
  595. "name": "scores_threshold",
  596. "type": "float32",
  597. "description": "Confidence threshold for NMS filtering",
  598. "visible": false
  599. },
  600. {
  601. "name": "iou_threshold",
  602. "type": "float32",
  603. "description": "Intersection over union overlap threshold, used in the NMS iterative elimination process where potential duplicates of detected items are ignored",
  604. "visible": false
  605. },
  606. {
  607. "name": "classes",
  608. "type": "int64",
  609. "description": "Number of NMS classes",
  610. "visible": false
  611. },
  612. {
  613. "name": "max_output_size",
  614. "type": "int64",
  615. "description": "Maximum number of proposals per class",
  616. "visible": false
  617. }
  618. ]
  619. },
  620. {
  621. "name": "normalization",
  622. "category": "Normalization",
  623. "description": "Normalization layer",
  624. "attributes": [
  625. {
  626. "name": "original_names",
  627. "type": "string[]",
  628. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  629. "visible": false
  630. },
  631. {
  632. "name": "elementwise_add",
  633. "type": "boolean",
  634. "description": "Whether elementwise addition is folded into the layer",
  635. "visible": false
  636. },
  637. {
  638. "name": "activation",
  639. "type": "string",
  640. "description": "Activation function name",
  641. "visible": false
  642. }
  643. ]
  644. },
  645. {
  646. "name": "proposal_generator",
  647. "description": "Proposal generator layer",
  648. "attributes": [
  649. {
  650. "name": "original_names",
  651. "type": "string[]",
  652. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  653. "visible": false
  654. }]
  655. },
  656. {
  657. "name": "reduce_l2",
  658. "description": "Reduce layer",
  659. "attributes": [
  660. {
  661. "name": "original_names",
  662. "type": "string[]",
  663. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  664. "visible": false
  665. }]
  666. },
  667. {
  668. "name": "reduce_max",
  669. "description": "Reduce Max layer",
  670. "attributes": [
  671. {
  672. "name": "original_names",
  673. "type": "string[]",
  674. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  675. "visible": false
  676. },
  677. {
  678. "name": "groups",
  679. "type": "int64",
  680. "description": "Number of groups input channels and output channels are divided into",
  681. "visible": false
  682. }
  683. ]
  684. },
  685. {
  686. "name": "reduce_sum",
  687. "description": "Reduce Sum layer",
  688. "attributes": [
  689. {
  690. "name": "original_names",
  691. "type": "string[]",
  692. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  693. "visible": false
  694. },
  695. {
  696. "name": "groups",
  697. "type": "int64",
  698. "description": "Number of groups input channels and output channels are divided into",
  699. "visible": false
  700. },
  701. {
  702. "name": "activation",
  703. "type": "string",
  704. "description": "Activation function name",
  705. "visible": false
  706. },
  707. {
  708. "name": "reduce_axes",
  709. "type": "int64[]",
  710. "description": "List of axes to reduce",
  711. "visible": false
  712. }
  713. ]
  714. },
  715. {
  716. "name": "resize",
  717. "category": "Tensor",
  718. "description": "Resize layer",
  719. "attributes": [
  720. {
  721. "name": "original_names",
  722. "type": "string[]",
  723. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  724. "visible": false
  725. },
  726. {
  727. "name": "resize_h_ratio_list",
  728. "type": "float32[]",
  729. "visible": true
  730. },
  731. {
  732. "name": "resize_w_ratio_list",
  733. "type": "float32[]",
  734. "visible": true
  735. },
  736. {
  737. "name": "resize_f_ratio_list",
  738. "type": "float32[]",
  739. "visible": true
  740. },
  741. {
  742. "name": "method",
  743. "type": "string",
  744. "description": "Resize method, either bilinear or nearest_neighbor",
  745. "visible": false
  746. },
  747. {
  748. "name": "resize_bilinear_pixels_mode",
  749. "type": "string",
  750. "description": "Bilinear resize variant, either half_pixels, align_corners, or disabled (where both align_corners and half_pixels are false)",
  751. "visible": false
  752. }
  753. ]
  754. },
  755. {
  756. "name": "shortcut",
  757. "description": "Shortcut layer",
  758. "attributes": [
  759. {
  760. "name": "original_names",
  761. "type": "string[]",
  762. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  763. "visible": false
  764. }]
  765. },
  766. {
  767. "name": "slice",
  768. "description": "Slice layer",
  769. "attributes": [
  770. {
  771. "name": "original_names",
  772. "type": "string[]",
  773. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  774. "visible": false
  775. },
  776. {
  777. "name": "height_slice",
  778. "type": "int64[]",
  779. "visible": false,
  780. "description": "Slice in the height axis (start, stop, step)"
  781. },
  782. {
  783. "name": "width_slice",
  784. "type": "int64[]",
  785. "visible": false,
  786. "description": "Slice in the width axis (start, stop, step)"
  787. },
  788. {
  789. "name": "features_slice",
  790. "type": "int64[]",
  791. "visible": false,
  792. "description": "Slice in the features axis (start, stop, step)"
  793. }
  794. ]
  795. },
  796. {
  797. "name": "softmax",
  798. "category": "Activation",
  799. "description": "Softmax layer",
  800. "attributes": [
  801. {
  802. "name": "original_names",
  803. "type": "string[]",
  804. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  805. "visible": false
  806. },
  807. {
  808. "name": "groups",
  809. "type": "int64",
  810. "description": "Number of groups input channels and output channels are divided into",
  811. "visible": false
  812. }
  813. ]
  814. },
  815. {
  816. "name": "space_to_depth",
  817. "description": "Space to depth layer",
  818. "attributes": [
  819. {
  820. "name": "original_names",
  821. "type": "string[]",
  822. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  823. "visible": false
  824. },
  825. {
  826. "name": "block_sizes",
  827. "type": "int64[]",
  828. "description": "Block size along each spatial axis",
  829. "visible": false
  830. },
  831. {
  832. "name": "space_to_depth_type",
  833. "type": "string",
  834. "description": "Space to depth variant, either classic_dcr (depth-column-row) classic_crd (column-row-depth), serial (used by Transformers patchify function), or focus (Yolov5-like)",
  835. "visible": false
  836. },
  837. {
  838. "name": "spatial_flatten_output",
  839. "type": "boolean",
  840. "description": "Whether to flatten the layer's output to one row",
  841. "visible": false
  842. }
  843. ]
  844. },
  845. {
  846. "name": "output_mux",
  847. "description": "Output muxer layer",
  848. "attributes": [
  849. {
  850. "name": "original_names",
  851. "type": "string[]",
  852. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  853. "visible": false
  854. }]
  855. },
  856. {
  857. "name": "concat",
  858. "category": "Tensor",
  859. "description": "Concatenation layer",
  860. "attributes": [
  861. {
  862. "name": "original_names",
  863. "type": "string[]",
  864. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  865. "visible": false
  866. },
  867. {
  868. "name": "concat_axis",
  869. "type": "int64",
  870. "description": "Axis to concatenate along, either features or spatial_w (which means the width axis)",
  871. "visible": false
  872. },
  873. {
  874. "name": "spatial_w_concat",
  875. "type": "boolean",
  876. "description": "Whether the concat operation is in the width dimension",
  877. "visible": false
  878. }
  879. ]
  880. },
  881. {
  882. "name": "matmul",
  883. "description": "Matrix multiplication layer",
  884. "attributes": [
  885. {
  886. "name": "original_names",
  887. "type": "string[]",
  888. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  889. "visible": false
  890. },
  891. {
  892. "name": "kernel_shape",
  893. "type": "int64[]",
  894. "label": "kernel",
  895. "description": "Shape of the kernel in Tensorflow convention (kernel height, kernel width, features in, features out)",
  896. "visible": true
  897. },
  898. {
  899. "name": "dynamic_weights",
  900. "type": "boolean",
  901. "description": "Whether the layer's weights are data driven",
  902. "visible": false
  903. },
  904. {
  905. "name": "transpose_matmul_input",
  906. "type": "boolean",
  907. "description": "Whether to transpose the width and the features axes of the layer's second input tensor",
  908. "visible": false
  909. }
  910. ]
  911. },
  912. {
  913. "name": "ew_add",
  914. "description": "Elementwise addition layer",
  915. "attributes": [
  916. {
  917. "name": "original_names",
  918. "type": "string[]",
  919. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  920. "visible": false
  921. },
  922. {
  923. "name": "activation",
  924. "type": "string",
  925. "description": "Activation function name",
  926. "visible": false
  927. }
  928. ]
  929. },
  930. {
  931. "name": "ew_div",
  932. "description": "Elementwise division layer",
  933. "attributes": [
  934. {
  935. "name": "original_names",
  936. "type": "string[]",
  937. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  938. "visible": false
  939. },
  940. {
  941. "name": "activation",
  942. "type": "string",
  943. "description": "Activation function name",
  944. "visible": false
  945. }
  946. ]
  947. },
  948. {
  949. "name": "ew_mult",
  950. "description": "Elementwise multiplication layer",
  951. "attributes": [
  952. {
  953. "name": "original_names",
  954. "type": "string[]",
  955. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  956. "visible": false
  957. },
  958. {
  959. "name": "activation",
  960. "type": "string",
  961. "description": "Activation function name",
  962. "visible": false
  963. }
  964. ]
  965. },
  966. {
  967. "name": "ew_sub",
  968. "description": "Elementwise subtraction layer",
  969. "attributes": [
  970. {
  971. "name": "original_names",
  972. "type": "string[]",
  973. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  974. "visible": false
  975. },
  976. {
  977. "name": "activation",
  978. "type": "string",
  979. "description": "Activation function name",
  980. "visible": false
  981. }
  982. ]
  983. },
  984. {
  985. "name": "demux",
  986. "description": "Demuxer layer",
  987. "attributes": [
  988. {
  989. "name": "original_names",
  990. "type": "string[]",
  991. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  992. "visible": false
  993. }]
  994. },
  995. {
  996. "name": "row_splitter",
  997. "description": "Splits the input tensor along the height axis",
  998. "attributes": [
  999. {
  1000. "name": "original_names",
  1001. "type": "string[]",
  1002. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  1003. "visible": false
  1004. }]
  1005. },
  1006. {
  1007. "name": "feature_splitter",
  1008. "description": "Splits the input tensor along the features axis",
  1009. "attributes": [
  1010. {
  1011. "name": "original_names",
  1012. "type": "string[]",
  1013. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  1014. "visible": false
  1015. }]
  1016. },
  1017. {
  1018. "name": "const_input",
  1019. "category": "Constant",
  1020. "description": "Constant input",
  1021. "attributes": [
  1022. {
  1023. "name": "original_names",
  1024. "type": "string[]",
  1025. "description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
  1026. "visible": false
  1027. }]
  1028. },
  1029. {
  1030. "name": "inv_pos",
  1031. "category": "Activation"
  1032. },
  1033. {
  1034. "name": "exp",
  1035. "category": "Activation"
  1036. },
  1037. {
  1038. "name": "silu",
  1039. "category": "Activation"
  1040. },
  1041. {
  1042. "name": "leaky",
  1043. "category": "Activation"
  1044. },
  1045. {
  1046. "name": "layer_normalization",
  1047. "category": "Normalization"
  1048. }
  1049. ]