/usr/local/lib64/python3.6/site-packages/caffe2/python/docs/__pycache__
Edit: /usr/local/lib64/python3.6/site-packages/caffe2/python/docs/__pycache__/github.cpython-36.pyc (5199B)
3
Eg
@ s d dl Z d dlZd dlmZ d dlmZmZ d dlmZmZ G dd deZ G dd deZ
d d
Z G d d d eZ G d
d deZ
G dd deZedkr
e jddZejdedd ej Zeejd6Zee
e Zej ejej e dj ej W dQ R X dS ) N)Markdown)OpDocGenerator DocUploader) OperatorDocOperatorEnginec @ s e Zd Zdd Zdd ZdS )GHOpDocUploaderc C s d S )N )selfr r E/usr/local/lib64/python3.6/site-packages/caffe2/python/docs/github.py__init__ s zGHOpDocUploader.__init__c C s t | d S )N)print)r content_bodyr r r
upload s zGHOpDocUploader.uploadN)__name__
__module__ __qualname__r r r r r r
r s r c @ s2 e Zd Zd ddZdd Zd
ddZdd d
Zd S )
GHMarkdown c C s | j dj|d |dd d S )Nz
{header} {text}
#)headertextT)addLineformat)r r hr r r
addHeader s zGHMarkdown.addHeaderc C sT | j d | j d | j d | j d | j d | j d | j d | j d d S )Nz---zdocid: operators-catalogztitle: Operators Catalogzlayout: operatorsz)permalink: /docs/operators-catalogue.htmlz* TOCz{:toc})r )r r r r
addDocHeader s
zGHMarkdown.addDocHeaderFc C s| | j t|dkst| jdjdd tt|d D | jdj|d x$|dd D ]}| jdj| q`W d S )Nr z | c S s g | ]}d qS )z
----------r ).0ir r r
' s z'GHMarkdown.addTable..r ) addLinebreaklenAssertionErrorr joinrange)r tablenoTitlerowr r r
addTable$ s &zGHMarkdown.addTablec C sn | j d xT|D ]L}| j d x2|D ]*}| j d | jdj|d | j d q$W | j d qW | j d d S ) Nz)addRawr r )r r$ r% r& r( r r r
addTableHTML, s
zGHMarkdown.addTableHTMLN)r )F)F)r r r r r r' r* r r r r
r s
r c C s@ | j } tjj|jd}dj|d}| jdj|d| | j S )NZcaffe2z5https://github.com/pytorch/pytorch/blob/master/{path})pathz{path})cloneosr+ relpathfiler ZaddLinkdump) formatterschemar+ Z
schemaLinkr r r
getCodeLink7 s r3 c @ s e Zd Zdd ZdS )GHOperatorEnginec C sR xL| j D ]@\}}|jdj| jdd |r
|jd |jdj|dd q
W d S )Nz{engine})engineTz on z{device})devicer )Z
getDeviceImpladdCoder r5 r) addEmphasis)r r1 r6 _r r r
generateDocA s
z GHOperatorEngine.generateDocN)r r r r: r r r r
r4 @ s r4 c @ s$ e Zd Zdd Zdd Zdd ZdS )
GHOperatorDocc C s* |j dd |j |jt|| j d S )NZCode )r r r) r3 r2 )r r1 r r r
generateCodeLinkJ s z GHOperatorDoc.generateCodeLinkc C sF |j }|r>|jdj|dd |jd |jdj|dd |j S )Nz{name})namer z{impl})implT)r, r8 r r) r7 r0 )r r1 r> r@ r r r
getInfoO s
zGHOperatorDoc.getInfoc C sJ |j | jd | jr<| j| | j| | j| |j n
|jd d S )N zNo schema documented yet.)r r> r2 r: ZgenerateInterfacer= ZaddBreakr )r r1 r r r
generateSchemaW s
z GHOperatorDoc.generateSchemaN)r r r r= rA rC r r r r
r; I s r; c @ s$ e Zd Zdd Zdd Zdd ZdS )GHOpDocGeneratorc C s t |||S )N)r; )r r> r2 priorityr r r
getOperatorDocc s z GHOpDocGenerator.getOperatorDocc C s t |S )N)r4 )r r> r r r
getOperatorEnginef s z"GHOpDocGenerator.getOperatorEnginec C sD | j j | j }x|D ]}|j| j qW | j| j j 7 _d S )N)r1 r Z getOperatorsrC r
r0 )r operatorsoperatorr r r
createBodyi s
zGHOpDocGenerator.createBodyN)r r r rF rG rJ r r r r
rD b s rD __main__z Operators catalog generator.) description catalog_pathz&operators-catalogue.md to write out to)typehelpwz Updated {}!) argparser- Z caffe2.python.docs.formatterr Z caffe2.python.docs.generatorr r r r r r r3 r4 r; rD r ArgumentParserparser add_argumentstr
parse_argsargsopenrM fpopsrJ writer
r r r r r r
s( !