MCPcopy Create free account
hub / github.com/pytorch/pytorch / generateInterface

Method generateInterface

caffe2/python/docs/generator.py:148–180  ·  view source on GitHub ↗
(self, formatter)

Source from the content-addressed store, hash-verified

146 formatter.addTable(table, (table == []))
147
148 def generateInterface(self, formatter):
149 def makeDesc(title, args):
150 f = formatter.clone()
151 f.addEmphasis(title, 1)
152 out = [(f.dump(), '')]
153 for arg in args:
154 f = formatter.clone()
155 if isinstance(arg, tuple):
156 name = arg[0]
157 if len(arg) > 1:
158 description = arg[1] or ''
159 else:
160 description = ''
161 else:
162 name = arg.name
163 description = arg.description or ''
164 f.addCode(name, inline=True)
165 out.append((f.dump(), description or ''))
166 return out
167
168 tuples = []
169
170 if self.schema.args:
171 tuples += makeDesc('Arguments', self.schema.args)
172
173 if self.schema.input_desc:
174 tuples += makeDesc('Inputs', self.schema.input_desc)
175
176 if self.schema.output_desc:
177 tuples += makeDesc('Outputs', self.schema.output_desc)
178
179 self.generateTable(formatter, tuples, None, 'Interface')
180 print("Generated interface for {}".format(self.name))
181
182 def generateCodeLink(self, formatter):
183 formatter.addHeader("Code", 3)

Callers 2

generateSchemaMethod · 0.95
generateSchemaMethod · 0.80

Calls 2

generateTableMethod · 0.95
formatMethod · 0.45

Tested by

no test coverage detected