Searched refs:dscale_index (Results 1 – 1 of 1) sorted by relevance
189 constexpr size_t dscale_index = 2; in InferImplFusedSparseAdam() local192 MS_EXCEPTION_IF_NULL(args_spec_list[dscale_index]); in InferImplFusedSparseAdam()196 auto dscale = args_spec_list[dscale_index]->Broaden(); in InferImplFusedSparseAdam()