Home
last modified time | relevance | path

Searched refs:multi_tensor_apply_for_fused_optimizer (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/
H A DFusedSgdKernel.mm38 multi_tensor_apply_for_fused_optimizer<3, 512>(kernel_name,
72 multi_tensor_apply_for_fused_optimizer<3, 512>(kernel_name,
129 multi_tensor_apply_for_fused_optimizer<2, 512>(kernel_name,
194 multi_tensor_apply_for_fused_optimizer<2, 512>(kernel_name,
H A DFusedAdamKernelImpl.mm30 multi_tensor_apply_for_fused_optimizer<4, 512>(kernel_name,
60 multi_tensor_apply_for_fused_optimizer<4, 512>(kernel_name,
H A DFusedAdamWKernelImpl.mm30 multi_tensor_apply_for_fused_optimizer<4, 512>(kernel_name,
60 multi_tensor_apply_for_fused_optimizer<4, 512>(kernel_name,
H A DFusedAdamAmsgradKernelImpl.mm32 multi_tensor_apply_for_fused_optimizer<5, 512>(kernel_name,
64 multi_tensor_apply_for_fused_optimizer<5, 512>(kernel_name,
H A DFusedAdamWAmsgradKernelImpl.mm32 multi_tensor_apply_for_fused_optimizer<5, 512>(kernel_name,
63 multi_tensor_apply_for_fused_optimizer<5, 512>(kernel_name,
H A DMultiTensorApply.h209 static void multi_tensor_apply_for_fused_optimizer(
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A Dfused_adam_impl.cu40 multi_tensor_apply_for_fused_optimizer<4>( in _fused_adam_cuda_impl_()
86 multi_tensor_apply_for_fused_optimizer<4>( in _fused_adam_cuda_impl_()
H A Dfused_adam_amsgrad_impl.cu45 multi_tensor_apply_for_fused_optimizer<5>( in _fused_adam_amsgrad_cuda_impl_()
96 multi_tensor_apply_for_fused_optimizer<5>( in _fused_adam_amsgrad_cuda_impl_()
H A Dfused_adamw_amsgrad_impl.cu46 multi_tensor_apply_for_fused_optimizer<5>( in _fused_adamw_amsgrad_cuda_impl_()
97 multi_tensor_apply_for_fused_optimizer<5>( in _fused_adamw_amsgrad_cuda_impl_()
H A Dfused_adamw_impl.cu41 multi_tensor_apply_for_fused_optimizer<4>( in _fused_adamw_cuda_impl_()
87 multi_tensor_apply_for_fused_optimizer<4>( in _fused_adamw_cuda_impl_()
H A DMultiTensorApply.cuh298 void multi_tensor_apply_for_fused_optimizer( in multi_tensor_apply_for_fused_optimizer() function