Searched refs:vec_contiguous (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | Blas.cpp | 782 const auto vec_contiguous = vec_stride == 0 ? vec.contiguous() : vec; in TORCH_IMPL_FUNC() local
784 vec_stride = std::max<int64_t>(vec_contiguous.stride(0), 1LL); in TORCH_IMPL_FUNC()
791 …, mat.size(1), alpha, mat.const_data_ptr<scalar_t>(), mat.stride(1), vec_contiguous.const_data_ptr… in TORCH_IMPL_FUNC()
797 …vec_contiguous.const_data_ptr<scalar_t>(), vec_stride, beta, result.mutable_data_ptr<scalar_t>(), … in TORCH_IMPL_FUNC()
803 …vec_contiguous.const_data_ptr<scalar_t>(), vec_stride, beta, result.mutable_data_ptr<scalar_t>(), … in TORCH_IMPL_FUNC()
|