Searched refs:input_scalar_t (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Normalization.cuh | 211 template <typename input_scalar_t, typename stat_scalar_t, typename stat_accscalar_t, bool train, t…
213 const GenericPackedTensorAccessor<const input_scalar_t, 3, RestrictPtrTraits, index_t> input, in batch_norm_transform_input_kernel() argument
214 GenericPackedTensorAccessor<input_scalar_t, 3, RestrictPtrTraits, index_t> output, in batch_norm_transform_input_kernel() argument
245 o[feature] = static_cast<input_scalar_t>(gamma * (i[feature] - mean) * invstd + beta); in batch_norm_transform_input_kernel()
268 template <typename VarTransform, typename input_scalar_t, typename stat_scalar_t, typename stat_acc…
270 const GenericPackedTensorAccessor<const input_scalar_t, 3, RestrictPtrTraits, index_t> input, in batch_norm_collect_statistics_kernel() argument
355 template <typename input_scalar_t, typename stat_scalar_t, typename stat_accscalar_t, typename inde…
357 const GenericPackedTensorAccessor<const input_scalar_t, 3, DefaultPtrTraits, index_t> input, in batch_norm_backward_kernel() argument
358 … const GenericPackedTensorAccessor<const input_scalar_t, 3, DefaultPtrTraits, index_t> grad_output, in batch_norm_backward_kernel()
359 GenericPackedTensorAccessor<input_scalar_t, 3, DefaultPtrTraits, index_t> grad_input, in batch_norm_backward_kernel() argument
[all …]
|
| H A D | ReflectionPad.cu | 163 template <typename input_scalar_t, typename output_scalar_t, typename F>
165 PackedTensorAccessor64<input_scalar_t, 5> input, in parallel_reflection_pad3d() argument
|