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 …]
|