Home
last modified time | relevance | path

Searched defs:grad_idx (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_conditional_accumulator.h149 const Tensor* grad_idx = std::get<0>(*grad); in AllocateAndAssignToAccumGradFunction() local
179 const Tensor* grad_idx = std::get<0>(*grad); in AddToAccumGradFunction() local
/external/tensorflow/tensorflow/core/ops/
Dtraining_ops.cc52 static Status HandleGradAndIndicesInputs(InferenceContext* c, int grad_idx, in HandleGradAndIndicesInputs()