Returns a signature for the given op, marking ragged args in bold.
(op, ragged_args)
| 520 | |
| 521 | |
| 522 | def _ragged_op_signature(op, ragged_args): |
| 523 | """Returns a signature for the given op, marking ragged args in bold.""" |
| 524 | op_name = tf_export.get_canonical_name_for_symbol(op) |
| 525 | argspec = tf_inspect.getfullargspec(op) |
| 526 | arg_names = argspec.args |
| 527 | |
| 528 | # Mark ragged arguments in bold. |
| 529 | for pos in ragged_args: |
| 530 | arg_names[pos] = '**' + arg_names[pos] + '**' |
| 531 | |
| 532 | # Add argument defaults. |
| 533 | for pos in range(-1, -len(argspec.defaults) - 1, -1): |
| 534 | arg_names[pos] += '=`{!r}`'.format(argspec.defaults[pos]) |
| 535 | |
| 536 | # Add varargs and keyword args |
| 537 | if argspec.varargs: |
| 538 | arg_names.append('*' + argspec.varargs) |
| 539 | if argspec.varkw: |
| 540 | arg_names.append('**' + argspec.varkw) |
| 541 | |
| 542 | return '* `tf.{}`({})'.format(op_name, ', '.join(arg_names)) |
| 543 | |
| 544 | |
| 545 | def _op_is_in_tf_version(op, version): |
no test coverage detected