Home
last modified time | relevance | path

Searched refs:src_dim_size (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/native/cuda/
DScatterGatherKernel.cu199 auto src_dim_size = ensure_nonempty_size(src, dim); in operator ()() local
201 auto index_size = is_scatter_like ? self_dim_size : src_dim_size; in operator ()()
257 auto src_dim_size = ensure_nonempty_size(src, dim); in operator ()() local
259 auto index_size = is_scatter_like ? self_dim_size : src_dim_size; in operator ()()
316 auto src_dim_size = ensure_nonempty_size(src, dim); in operator ()() local
318 auto index_size = is_scatter_like ? self_dim_size : src_dim_size; in operator ()()
/external/pytorch/aten/src/ATen/native/cpu/
DScatterGatherKernel.cpp287 auto src_dim_size = ensure_nonempty_size(src, dim); in operator ()() local
289 auto index_upper_bound = is_scatter_like ? self_dim_size : src_dim_size; in operator ()()
383 auto src_dim_size = ensure_nonempty_size(src, dim); in operator ()() local
385 auto index_upper_bound = is_scatter_like ? self_dim_size : src_dim_size; in operator ()()
478 auto src_dim_size = ensure_nonempty_size(src, dim); in operator ()() local
480 auto index_upper_bound = is_scatter_like ? self_dim_size : src_dim_size; in operator ()()
574 auto src_dim_size = ensure_nonempty_size(src, dim); in operator ()() local
576 auto index_upper_bound = is_scatter_like ? self_dim_size : src_dim_size; in operator ()()