Home
last modified time | relevance | path

Searched defs:bitwise_right_shift (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DBinaryOps.cpp280 TORCH_META_FUNC2(bitwise_right_shift, Tensor) ( in TORCH_META_FUNC2() argument
1373 Tensor bitwise_right_shift(const Tensor& self, const Scalar& other) { in bitwise_right_shift() function
1381 Tensor bitwise_right_shift(const Scalar& self, const Tensor& other) { in bitwise_right_shift() function
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dcpp.py910 def bitwise_right_shift(a, b): member in CppOverrides
1224 def bitwise_right_shift(a, b): member in CppVecOverrides
H A Dhalide.py415 def bitwise_right_shift(a, b): member in HalideOverrides
H A Dtriton.py918 def bitwise_right_shift(a, b): member in TritonOverrides
H A Dcommon.py847 def bitwise_right_shift(x, y): member in OpOverrides
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dops_handler.py423 def bitwise_right_shift(self, x0: T, x1: T) -> T: member in OpsHandler
H A Dlowering.py5840 bitwise_right_shift = register_pointwise(aten.bitwise_right_shift) variable
/aosp_15_r20/external/pytorch/torch/_refs/
H A D__init__.py1136 def bitwise_right_shift(a: TensorLikeType, b: TensorLikeType) -> TensorLikeType: function