def main(argv):
cpu_ops = ops.cpu_ops()
gpu_ops = ops.gpu_ops()
mix_ops = ops.mixed_ops()
support_ops = ops.support_ops()
all_ops = cpu_ops.union(gpu_ops).union(mix_ops).union(support_ops)
link_string = '_'
op_name_max_len = len(max(all_ops, key=len)) + len(link_string)
name_bar = op_name_max_len * '='
formater = '{:{c}<{op_name_max_len}} {:{c}^6} {:{c}^6} {:{c}^6} {:{c}^7} {:{c}^9}\n'
doc_table = ''
doc_table += 'Below table lists all available operators and devices they can operate on.\n\n'
doc_table += '.. |v| image:: images/tick.gif\n'
doc_table += formater.format('', '', '', '', '', '', op_name_max_len = op_name_max_len, c='=')
doc_table += formater.format('Operator name', 'CPU', 'GPU', 'Mixed', 'Support', 'Sequences', op_name_max_len = op_name_max_len, c=' ')
doc_table += formater.format('', '', '', '', '', '', op_name_max_len = op_name_max_len, c='=')
for op in sorted(all_ops, key=lambda v: str(v).lower()):
schema = b.GetSchema(op)
is_cpu = '|v|' if op in cpu_ops else ''
is_gpu = '|v|' if op in gpu_ops else ''
is_mixed = '|v|' if op in mix_ops else ''
is_support = '|v|' if op in support_ops else ''
supports_seq = '|v|' if schema.AllowsSequences() or schema.IsSequenceOperator() else ''
op_string = op + link_string
op_doc = formater.format(op_string, is_cpu, is_gpu, is_mixed, is_support, supports_seq, op_name_max_len = op_name_max_len, c=' ')
doc_table += op_doc
doc_table += formater.format('', '', '', '', '', '', op_name_max_len = op_name_max_len, c='=')
with open(argv[0], 'w') as f:
f.write(doc_table)
def main(out_filename):
cpu_ops = ops.cpu_ops()
gpu_ops = ops.gpu_ops()
mix_ops = ops.mixed_ops()
all_ops = cpu_ops.union(gpu_ops).union(mix_ops)
longest_module = max(ops_modules.keys(), key = len)
link_formatter = ':meth:`{op} <{module}.{op}>`'
op_name_max_len = len(link_formatter.format(op = "", module = longest_module)) + \
2 * len(max(all_ops, key=len))
name_bar = op_name_max_len * '='
formater = '{:{c}<{op_name_max_len}} {:{c}^6} {:{c}^6} {:{c}^7} {:{c}^9} {:{c}^10}\n'
doc_table = ''
doc_table += '.. |v| image:: images/tick.gif\n'
doc_table += '\n'
doc_table += formater.format('', '', '', '', '', '', op_name_max_len = op_name_max_len, c='=')
doc_table += formater.format('Operator name', 'CPU', 'GPU', 'Mixed', 'Sequences', 'Volumetric', op_name_max_len = op_name_max_len, c=' ')
doc_table += formater.format('', '', '', '', '', '', op_name_max_len = op_name_max_len, c='=')
for op in sorted(all_ops, key=name_sort):
schema = b.GetSchema(op)
op_full_name, submodule, op_name = ops._process_op_name(op)
is_cpu = '|v|' if op in cpu_ops else ''
is_gpu = '|v|' if op in gpu_ops else ''
is_mixed = '|v|' if op in mix_ops else ''
supports_seq = '|v|' if schema.AllowsSequences() or schema.IsSequenceOperator() else ''
volumetric = '|v|' if schema.SupportsVolumetric() else ''
for (module_name, module) in ops_modules.items():
m = module
for part in submodule:
m = getattr(m, part, None)
if m is None:
break
if m is not None and hasattr(m, op_name):
submodule_str = ".".join([*submodule])
op_string = link_formatter.format(op = op_full_name, module = module_name)
op_doc = formater.format(op_string, is_cpu, is_gpu, is_mixed, supports_seq, volumetric, op_name_max_len = op_name_max_len, c=' ')
doc_table += op_doc
doc_table += formater.format('', '', '', '', '', '', op_name_max_len = op_name_max_len, c='=')
with open(out_filename, 'w') as f:
f.write(doc_table)
'nvidia.dali.ops': nvidia.dali.ops,
'nvidia.dali.plugin.pytorch': nvidia.dali.plugin.pytorch,
'nvidia.dali.plugin.numba.experimental':
nvidia.dali.plugin.numba.experimental
}
# Some operators might have a different module for the fn wrapper
module_mapping = {
'nvidia.dali.plugin.pytorch':
'nvidia.dali.plugin.pytorch.fn',
'nvidia.dali.plugin.numba.experimental':
'nvidia.dali.plugin.numba.fn.experimental'
}
cpu_ops = ops.cpu_ops()
gpu_ops = ops.gpu_ops()
mix_ops = ops.mixed_ops()
all_ops = cpu_ops.union(gpu_ops).union(mix_ops)
link_formatter = ':meth:`{op} <{module}.{op}>`'
def to_fn_name(full_op_name):
tokens = full_op_name.split('.')
tokens[-1] = fn._to_snake_case(tokens[-1])
return '.'.join(tokens)
def to_fn_module(module_name):
if module_name in module_mapping:
return module_mapping[module_name]
else:
def main(argv):
cpu_ops = ops.cpu_ops()
gpu_ops = ops.gpu_ops()
mix_ops = ops.mixed_ops()
all_ops = cpu_ops.union(gpu_ops).union(mix_ops)
longest_module = max(ops_modules.keys(), key=len)
link_formatter = ':meth:`{op} <{module}.{op}>`'
op_name_max_len = len(link_formatter.format(op = "", module = longest_module)) + \
2 * len(max(all_ops, key=len))
name_bar = op_name_max_len * '='
formater = '{:{c}<{op_name_max_len}} {:{c}^6} {:{c}^6} {:{c}^7} {:{c}^9} {:{c}^10}\n'
doc_table = ''
doc_table += '.. |v| image:: images/tick.gif\n'
doc_table += '\n'
doc_table += formater.format('',
'',
'',
'',
'',
'',
op_name_max_len=op_name_max_len,
c='=')
doc_table += formater.format('Operator name',
'CPU',
'GPU',
'Mixed',
'Sequences',
'Volumetric',
op_name_max_len=op_name_max_len,
c=' ')
doc_table += formater.format('',
'',
'',
'',
'',
'',
op_name_max_len=op_name_max_len,
c='=')
for op in sorted(all_ops, key=lambda v: str(v).lower()):
schema = b.GetSchema(op)
is_cpu = '|v|' if op in cpu_ops else ''
is_gpu = '|v|' if op in gpu_ops else ''
is_mixed = '|v|' if op in mix_ops else ''
supports_seq = '|v|' if schema.AllowsSequences(
) or schema.IsSequenceOperator() else ''
volumetric = '|v|' if schema.SupportsVolumetric() else ''
for (module_name, module) in ops_modules.items():
if hasattr(module, op):
op_string = link_formatter.format(op=op, module=module_name)
op_doc = formater.format(op_string,
is_cpu,
is_gpu,
is_mixed,
supports_seq,
volumetric,
op_name_max_len=op_name_max_len,
c=' ')
doc_table += op_doc
doc_table += formater.format('',
'',
'',
'',
'',
'',
op_name_max_len=op_name_max_len,
c='=')
with open(argv[0], 'w') as f:
f.write(doc_table)