Searched refs:_move_memory_if_cuda_input (Results 1 – 1 of 1) sorted by relevance
2292 inline Tensor _move_memory_if_cuda_input( in _move_memory_if_cuda_input() function2316 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()