Searched refs:scalar_end (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | RangeFactories.cu | 88 scalar_t scalar_end = end.to<scalar_t>(); in linspace_cuda_out() local
90 … float step = (static_cast<float>(scalar_end) - static_cast<float>(scalar_start)) / (steps - 1); in linspace_cuda_out()
92 …gpu_kernel_with_index(r, [scalar_start, scalar_end, steps, step, halfway]GPU_LAMBDA(int64_t ind) -… in linspace_cuda_out()
97 return scalar_end - step * (steps - ind - 1); in linspace_cuda_out()
103 scalar_t scalar_end = end.to<scalar_t>(); in linspace_cuda_out() local
104 scalar_t step = (scalar_end - scalar_start) / static_cast<scalar_t>(steps - 1); in linspace_cuda_out()
106 …gpu_kernel_with_index(r, [scalar_start, scalar_end, steps, step, halfway]GPU_LAMBDA(int64_t ind) -… in linspace_cuda_out()
111 return scalar_end - step * (steps - ind - 1); in linspace_cuda_out()
144 scalar_t scalar_end = end.to<scalar_t>(); in logspace_cuda_out() local
145 float step = static_cast<float>(scalar_end - scalar_start) / (steps - 1); in logspace_cuda_out()
[all …]
|
| /external/pytorch/aten/src/ATen/native/ |
| D | RangeFactories.cpp | 112 scalar_t scalar_end = end.to<scalar_t>(); in logspace_out() local
114 scalar_t step = (scalar_end - scalar_start) / static_cast<scalar_t>(steps - 1); in logspace_out()
122 … data_ptr[i] = std::pow(scalar_base, scalar_end - (step * static_cast<scalar_t>(steps - i - 1))); in logspace_out()
131 scalar_t scalar_end = end.to<scalar_t>(); in logspace_out() local
133 double step = static_cast<double>(scalar_end - scalar_start) / (steps - 1); in logspace_out()
140 data_ptr[i] = std::pow(scalar_base, scalar_end - step * (steps - i - 1)); in logspace_out()
|
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | RangeFactoriesKernel.cpp | 45 …kernel(TensorIterator& iter, const Scalar& scalar_start, const Scalar& scalar_end, int64_t steps) { in linspace_kernel() argument
50 const scalar_t end = scalar_end.to<scalar_t>(); in linspace_kernel()
|