Home
last modified time | relevance | path

Searched refs:kGroupByDims (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_softmax_op.cc89 const ArraySlice<int64> kGroupByDims = kReorderDims.subspan(0, rank - 1); in Compute() local
97 for (const auto &g : st.group(kGroupByDims)) { in Compute()