Home
last modified time | relevance | path

Searched defs:all_aligned (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DForeachFunctors.cuh28 bool all_aligned = true; in init_args() local
48 bool all_aligned = true; in init_args() local
67 bool all_aligned = true; in init_args() local
120 const bool all_aligned, in binary_op_scalar()
162 const bool all_aligned, in pointwise_op_scalar()
220 const bool all_aligned = in operator ()() local
243 const bool all_aligned = in operator ()() local
268 const bool all_aligned = in operator ()() local
321 const bool all_aligned = in operator ()() local
374 const auto all_aligned = in operator ()() local
[all …]
H A Dfused_adam_utils.cuh144 const bool all_aligned{ in operator ()() local
H A DFusedSgdKernel.cu85 const auto all_aligned{ in operator ()() local
H A DForeachBinaryOpList.cu351 const bool all_aligned{is_aligned(src_ptr) && is_aligned(self_ptr)}; in operator ()() local