Home
last modified time | relevance | path

Searched refs:elementwise_unary_scalar_wrapper (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_refs/nn/functional/
H A D__init__.py21 elementwise_unary_scalar_wrapper,
884 @elementwise_unary_scalar_wrapper
1007 @elementwise_unary_scalar_wrapper
1043 @elementwise_unary_scalar_wrapper
/aosp_15_r20/external/pytorch/torch/_prims_common/
H A Dwrappers.py411 def elementwise_unary_scalar_wrapper(fn: Callable) -> Callable: function
/aosp_15_r20/external/pytorch/torch/_refs/
H A D__init__.py48 elementwise_unary_scalar_wrapper,
470 @elementwise_unary_scalar_wrapper