/usr/local/lib64/python3.6/site-packages/caffe2/python/docs/__pycache__
Edit: /usr/local/lib64/python3.6/site-packages/caffe2/python/docs/__pycache__/generator.cpython-36.pyc (7601B)
3
EgR
@ s d dl Z d dlZd dlmZmZ d dlmZ d dlmZm Z ej
j Z G dd de Z
G dd de ZG d d
d
eZG d d d e ZG d
d de Zedkre jddZejdedd ej Zeejd(Zee e
Z e j ej e j W dQ R X dS ) N)core workspace)Markdown) viewitems
viewvaluesc @ s e Zd Zdd Zdd ZdS ) DocUploaderc C s d S )N )selfr r H/usr/local/lib64/python3.6/site-packages/caffe2/python/docs/generator.py__init__ s zDocUploader.__init__c C s d S )Nr )r textr r r
upload s zDocUploader.uploadN)__name__
__module__ __qualname__r r
r r r r
r s r c @ s$ e Zd Zdd Zdd Zdd ZdS ) DocGeneratorc C s || _ || _d| _d S )N ) formatteruploader content_body)r r r r r r
r s zDocGenerator.__init__c C s d S )Nr )r r r r
create_body s zDocGenerator.create_bodyc C s | j j| j d S )N)r r
r )r r r r
update! s zDocGenerator.updateN)r r r r r r r r r r
r s r c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
OpDocGeneratorc C s t |||S )N) OperatorDoc)r nameschemapriorityr r r
getOperatorDoc& s z OpDocGenerator.getOperatorDocc C s t |S )N)OperatorEngine)r r r r r
getOperatorEngine) s z OpDocGenerator.getOperatorEnginec C s i | _ i | _dd }xtj D ]}tj|}|r\|tjj|j }| j
|||}|| j |< q |j dd kr| j |}|j
| jkr| j|j
j| q|g| j|j
< q d}| j
|||| j |< q W x0t| jD ]"\}}|| j kr| j | j| qW tt| j dd dS )
Nc S s4 | dkr dS d| j dkr dS d| j dkr0dS dS ) Nzcaffe2/caffe2/operatorsr contrib/ Z experiments )split)xr r r
filePriority2 s z1OpDocGenerator.getOperators.
.filePriority_ENGINE_r$ c S s | j | jfS )N)r r )opr r r
V s z-OpDocGenerator.getOperators..)key) operatorsenginesr Z_GetRegisteredOperatorsOpSchemagetospathdirnamefiler findr base_op_nameappendr
addEnginessortedr )r r' r r r operatorenginer/ r r r
getOperators, s,
zOpDocGenerator.getOperatorsc C s: | j }x|D ]}|j| j qW | j| jj 7 _d S )N)r= generateSchemar r dump)r r. r; r r r
createBodyY s
zOpDocGenerator.createBodyN)r r r r r r= r@ r r r r
r % s -r c @ s$ e Zd Zdd Zdd Zdd ZdS )r c C s || _ |jdd\| _| _d S )Nr( r$ )op_namer% r7 r< )r r r r r
r c s zOperatorEngine.__init__c C sJ g }x@dt j| jfdt j| jfgD ] \}}|s4q&|j||f q&W |S )NCPUZCUDA)r0 get_cpu_implrA
get_cuda_implr8 )r ZdeviceImplListdeviceimplr r r
getDeviceImplg s z OperatorEngine.getDeviceImplc C s2 x,| j D ] \}}|jdj| j||d q
W d S )Nz {engine} on {device}: {impl})r< rE rF )rG addLineformatr< )r r rE rF r r r
generateDocp s
zOperatorEngine.generateDocN)r r r r rG rJ r r r r
r b s r c @ s\ e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
dd Z dd Z dS )r c C s, || _ || _|| _tdj| j g | _d S )NzGathering docs for {}...)r r r printrI r/ )r r r r r r r
r y s
zOperatorDoc.__init__c C s
|| _ d S )N)r/ )r r/ r r r
r9 s zOperatorDoc.addEnginesc C s. | j jr |j| j j |j n
|jd d S )NzNo documentation yet.)r docZ parseAndAdd addLinebreakrH )r r r r r
rJ s
zOperatorDoc.generateDocc C sZ |rV|r|j |d g }|r"|g}x"|D ]\}}|j||p.makeDesc ArgumentsZInputsZOutputsZ InterfacezGenerated interface for {})r rY Z
input_descZ output_descrR rK rI r )r r r] rO r )r r
generateInterface s z OperatorDoc.generateInterfacec C s& |j dd |j |j| jj d S )NZCoder# )rN rM Z addCodeLinkr r5 )r r r r r
generateCodeLink s z OperatorDoc.generateCodeLinkc C s d S )Nr )r r r rF r r r
getInfo s zOperatorDoc.getInfoc C sP |j dd | j|dtj| j| j|dtj| jg}|jdd |D d S )NZDevicesr# rB ZGPUc S s g | ] }|r|qS r r ).0ir r r
s z/OperatorDoc.generateDevices..)rN ra r0 rC r rD ZaddList)r r Zdevicesr r r
generateDevices s zOperatorDoc.generateDevicesc C s8 t | jsd S |jdd x| jD ]}|j| q"W d S )NZEnginesr# )rW r/ rN rJ )r r r< r r r
generateEngines s
zOperatorDoc.generateEnginesc C sh |j | jd | jrP| j| | j| | j| | j| | j| |j n|j d | j| d S )Nr" zNo schema documented yet.)
rN r r rJ r_ r` re rf ZaddBreakrH )r r r r r
r> s
zOperatorDoc.generateSchemaN)
r r r r r9 rJ rR r_ r` ra re rf r> r r r r
r x s "
r __main__z Operators catalog generator.)rX catalog_pathz&operators-catalogue.md to write out to)typehelpw) argparser2 Z
caffe2.pythonr r Z caffe2.python.docs.formatterr Z future.utilsr r Cr0 objectr r r r r r ArgumentParserparser add_argumentstr
parse_argsrY openrh fpopsr@ writer r r r r
s&
=f