Searched defs:grad_idx (Results 1 – 2 of 2) sorted by relevance
149 const Tensor* grad_idx = std::get<0>(*grad); in AllocateAndAssignToAccumGradFunction() local179 const Tensor* grad_idx = std::get<0>(*grad); in AddToAccumGradFunction() local
52 static Status HandleGradAndIndicesInputs(InferenceContext* c, int grad_idx, in HandleGradAndIndicesInputs()