Searched defs:tensor_args (Results 1 – 2 of 2) sorted by relevance
120 def f_wrapper(*tensor_args):
273 def ragged_dispatch(original_op, tensor_args): argument