Searched defs:reduce_ (Results 1 – 4 of 4) sorted by relevance
/aosp_15_r20/external/mesa3d/src/compiler/nir/ |
H A D | nir_opcodes.py | 183 def reduce_(src0, src1): function 594 def reduce_(src0, src1): function
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/ |
H A D | softmax.h | 48 __device__ __forceinline__ void reduce_(Tensor<Engine0, Layout0> const& tensor, Tensor<Engine1, Lay… in reduce_() function
|
/aosp_15_r20/external/perfetto/src/trace_redaction/ |
H A D | collect_frame_cookies_unittest.cc | 233 ReduceFrameCookies reduce_; member in perfetto::trace_redaction::__anona767409b0111::ReduceFrameCookiesTest
|
/aosp_15_r20/external/pytorch/torch/ |
H A D | _meta_registrations.py | 4897 def get_operator_enum(reduce_, use_new_options=False): argument 4994 def scatter_meta_impl(self, dim, index, src=None, reduce_=None, use_new_options=False): argument
|