Searched defs:param_device (Results 1 – 3 of 3) sorted by relevance
41 std::optional<int64_t> param_device; in parameters_to_vector() local61 std::optional<int64_t> param_device; in vector_to_parameters() local
113 Device param_device = params[0].device(); in _fused_adam_kernel_cuda_() local
114 Device param_device = params[0].device(); in _fused_adamw_kernel_cuda_() local