Returns a string listing operators that have dispathers registered.
(tf_version=1)
| 553 | |
| 554 | |
| 555 | def ragged_op_list(tf_version=1): |
| 556 | """Returns a string listing operators that have dispathers registered.""" |
| 557 | lines = [] |
| 558 | for op in _UNARY_ELEMENTWISE_OPS + _UNARY_LIST_ELEMENTWISE_OPS: |
| 559 | if _op_is_in_tf_version(op, tf_version): |
| 560 | lines.append(_ragged_op_signature(op, [0])) |
| 561 | for op in _BINARY_ELEMENTWISE_OPS: |
| 562 | if _op_is_in_tf_version(op, tf_version): |
| 563 | lines.append(_ragged_op_signature(op, [0, 1])) |
| 564 | for op, _, ragged_args in _RAGGED_DISPATCH_OPS: |
| 565 | if _op_is_in_tf_version(op, tf_version): |
| 566 | arginfos = _get_arg_infos(op, ragged_args) |
| 567 | ragged_args = [arginfo.position for arginfo in arginfos] |
| 568 | lines.append(_ragged_op_signature(op, ragged_args)) |
| 569 | return ('\n\n### Additional ops that support `RaggedTensor`\n\n' |
| 570 | 'Arguments that accept `RaggedTensor`s are marked in **bold**.\n\n' + |
| 571 | '\n'.join(sorted(lines)) + 'n') |
| 572 | |
| 573 | |
| 574 | register_dispatchers() |
nothing calls this directly
no test coverage detected