Searched refs:var_first_dim_size (Results 1 – 4 of 4) sorted by relevance
36 const auto var_first_dim_size = input_params->var_first_dim_size_; in ComputeProximalAdagrad() local40 if (index < 0 || LongToSize(index) >= var_first_dim_size) { in ComputeProximalAdagrad()
39 const auto var_first_dim_size = input_params->var_first_dim_size_; in ComputeLazyAdam() local43 if (index < 0 || LongToSize(index) >= var_first_dim_size) { in ComputeLazyAdam()
38 const auto var_first_dim_size = input_params->var_first_dim_size_; in ComputeFtrl() local42 if (index < 0 || LongToSize(index) >= var_first_dim_size) { in ComputeFtrl()
37 const auto var_first_dim_size = input_params->var_first_dim_size_; in ComputeAdam() local41 if (index < 0 || LongToSize(index) >= var_first_dim_size) { in ComputeAdam()