Home
last modified time | relevance | path

Searched defs:self_v (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseUnaryOps.cpp224 const auto self_v = [&self]() { in threshold_backward_sparse() local
248 auto self_v = [&self]() { in threshold_backward_sparse_out() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DMultinomialKernel.cu341 auto self_v = inputSize == 1 ? self.view({numDist, numCategories}) : self; in multinomial_with_replacement_kernel_impl() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/xpu/
H A DBlas.cpp352 Tensor self_v; in addmv_out() local