Searched refs:group_indices (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | set_kernels.cc | 160 const auto& group_indices = it->first; in OutputSparseTensor() local
162 ctx, group_indices.size() == output_shape.dims() - 1, in OutputSparseTensor()
163 errors::Internal("Invalid number of indices ", group_indices.size(), in OutputSparseTensor()
173 for (int32 i = 0; i < group_indices.size(); ++i) { in OutputSparseTensor()
174 out_indices_mat(value_index, i) = group_indices[i]; in OutputSparseTensor()
176 out_indices_mat(value_index, group_indices.size()) = group_value_index; in OutputSparseTensor()
205 const std::vector<int64>& group_indices, in PopulateFromDenseGroup() argument
207 OP_REQUIRES(ctx, group_indices.size() == input_strides.size() - 1, in PopulateFromDenseGroup()
208 errors::Internal("group_indices.size ", group_indices.size(), in PopulateFromDenseGroup()
214 group_indices.begin(), group_indices.end(), input_strides.begin(), 0LL); in PopulateFromDenseGroup()
[all …]
|