Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/sparse/
Dsparse_tensor.h197 GroupIterable group(const VarDimArray& group_ix) const { in group() argument
198 DCHECK_LE(group_ix.size(), dims_); in group()
199 for (std::size_t di = 0; di < group_ix.size(); ++di) { in group()
200 DCHECK_GE(group_ix[di], 0) << "Group dimension out of range"; in group()
201 DCHECK_LT(group_ix[di], dims_) << "Group dimension out of range"; in group()
202 DCHECK_EQ(group_ix[di], order_[di]) in group()
205 return GroupIterable(ix_, vals_, dims_, group_ix); in group()
/external/tensorflow/tensorflow/core/kernels/
Dset_kernels.cc272 VarDimArray group_ix = set_st.order().subspan(0, set_st.order().size() - 1); in Compute() local
274 for (const auto& group : set_st.group(group_ix)) { in Compute()