Searched refs:set_st (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | set_kernels.cc | 251 sparse::SparseTensor set_st; in Compute() local
253 SparseTensorFromContext(ctx, 0, validate_indices_, &set_st)); in Compute()
254 OP_REQUIRES_OK(ctx, set_st.IndicesValid()); in Compute()
259 OP_REQUIRES_OK(ctx, GroupShape(set_st.shape(), &output_shape)); in Compute()
272 VarDimArray group_ix = set_st.order().subspan(0, set_st.order().size() - 1); in Compute()
274 for (const auto& group : set_st.group(group_ix)) { in Compute()
275 PopulateFromSparseGroup<T>(ctx, group, set_st.shape(), &group_set); in Compute()
|