Home
last modified time | relevance | path

Searched refs:ensure_tuple (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/torch/csrc/autograd/
Dpython_function.h80 inline bool ensure_tuple(THPObjectPtr& obj) { in ensure_tuple() function
Dpython_function.cpp153 ensure_tuple(r); in apply()
251 ensure_tuple(r); in defer_to_dynamo()
672 ensure_tuple(r); in _wrap_outputs()
1088 bool unpack_output = ensure_tuple(raw_output); in process_outputs()
/external/pytorch/test/
Dtest_sort_and_select.py927 def ensure_tuple(x): function
935 ret = ensure_tuple(
958 def ensure_tuple(x): function
1010 ret = ensure_tuple(
Dtest_mps.py4116 def ensure_tuple(x): function
4124 ret = ensure_tuple(f(x, return_inverse=return_inverse, return_counts=return_counts))
4142 def ensure_tuple(x): function
4177 … ret = ensure_tuple(f(x, return_inverse=return_inverse, return_counts=return_counts))