Home
last modified time | relevance | path

Searched defs:param_device (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/utils/
H A Dconvert_parameters.h41 std::optional<int64_t> param_device; in parameters_to_vector() local
61 std::optional<int64_t> param_device; in vector_to_parameters() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DFusedAdamKernel.cu113 Device param_device = params[0].device(); in _fused_adam_kernel_cuda_() local
H A DFusedAdamWKernel.cu114 Device param_device = params[0].device(); in _fused_adamw_kernel_cuda_() local