Searched refs:kSparseApplyLazyAdamWorkspaceSize (Results 1 – 1 of 1) sorted by relevance
25 constexpr size_t kSparseApplyLazyAdamWorkspaceSize = 4; variable175 CHECK_KERNEL_WORKSPACE_SIZE(workspace.size(), kSparseApplyLazyAdamWorkspaceSize, kernel_name_); in Launch()