Searched refs:ragged_op (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_dispatch.py | 228 def __init__(self, original_op, ragged_op, ragged_args): argument
230 ragged_arg_names = tf_inspect.getfullargspec(ragged_op)[0]
234 (original_op, ragged_op, op_arg_names, ragged_arg_names))
235 self._ragged_op = ragged_op
236 self._ragged_args = _get_arg_infos(ragged_op, ragged_args)
275 def decorator(ragged_op): argument
276 dispatch.RaggedDispatcher(original_op, ragged_op,
278 return ragged_op
487 for (original_op, ragged_op, args) in _RAGGED_DISPATCH_OPS:
488 RaggedDispatcher(original_op, ragged_op, args).register(original_op)
|