Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DLinearAlgebra.cpp2292 inline Tensor _move_memory_if_cuda_input( in _move_memory_if_cuda_input() function
2316 return _move_memory_if_cuda_input(tensor, in); in _blob_to_Tensor()
2465 bs = _move_memory_if_cuda_input(bs, A); in compute_T12()
2537 bs = _move_memory_if_cuda_input(bs, A); in compute_T18()
2673 auto idx_to_device = _move_memory_if_cuda_input( in mexp_impl()
2686 auto idx_to_device = _move_memory_if_cuda_input( in mexp_impl()