Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/
H A Dfunctional.py10 from torch._jit_internal import _overload as overload, boolean_dispatch
1097 _return_inverse_false = boolean_dispatch(
1107 _return_inverse_true = boolean_dispatch(
1120 unique = boolean_dispatch(
1181 _consecutive_return_inverse_false = boolean_dispatch(
1191 _consecutive_return_inverse_true = boolean_dispatch(
1204 unique_consecutive = boolean_dispatch(
2191 lu = boolean_dispatch(
H A D_jit_internal.py599 def boolean_dispatch( function
H A Doverrides.py238 torch._jit_internal.boolean_dispatch,
/aosp_15_r20/external/pytorch/torch/nn/
H A Dfunctional.py13 boolean_dispatch,
537 fractional_max_pool2d = boolean_dispatch(
653 fractional_max_pool3d = boolean_dispatch(
743 max_pool1d = boolean_dispatch(
833 max_pool2d = boolean_dispatch(
923 max_pool3d = boolean_dispatch(
1230 adaptive_max_pool1d = boolean_dispatch(
1286 adaptive_max_pool2d = boolean_dispatch(
1343 adaptive_max_pool3d = boolean_dispatch(