Searched defs: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 | 213 const GenericPackedTensorAccessor<const input_scalar_t, 3, RestrictPtrTraits, index_t> input, in batch_norm_transform_input_kernel()
214 GenericPackedTensorAccessor<input_scalar_t, 3, RestrictPtrTraits, index_t> output, in batch_norm_transform_input_kernel()
270 const GenericPackedTensorAccessor<const input_scalar_t, 3, RestrictPtrTraits, index_t> input, in batch_norm_collect_statistics_kernel()
357 const GenericPackedTensorAccessor<const input_scalar_t, 3, DefaultPtrTraits, index_t> input, in batch_norm_backward_kernel()
359 GenericPackedTensorAccessor<input_scalar_t, 3, DefaultPtrTraits, index_t> grad_input, in batch_norm_backward_kernel()
474 const GenericPackedTensorAccessor<input_scalar_t, 3, DefaultPtrTraits, index_t> input, in batch_norm_backward_reduce_kernel()
509 const GenericPackedTensorAccessor<input_scalar_t, 3, DefaultPtrTraits, index_t> input, in batch_norm_backward_elemt_kernel_impl()
547 const GenericPackedTensorAccessor<input_scalar_t, 3, DefaultPtrTraits, index_t> input, in batch_norm_backward_elemt_kernel()
569 const GenericPackedTensorAccessor<input_scalar_t, 3, DefaultPtrTraits, index_t> input, in batch_norm_backward_elemt_kernel()
|
| H A D | ReflectionPad.cu | 165 PackedTensorAccessor64<input_scalar_t, 5> input, in parallel_reflection_pad3d()
|