Searched refs:group_sets (Results 1 – 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | set_kernels.cc | 441 std::map<std::vector<int64>, std::set<T>> group_sets; in ComputeDenseToDense() local 462 group_sets[group_indices] = group_set; in ComputeDenseToDense() 474 OutputSparseTensor<T>(ctx, output_shape, num_result_values, group_sets); in ComputeDenseToDense() 497 std::map<std::vector<int64>, std::set<T>> group_sets; in ComputeDenseToSparse() local 544 group_sets[group_indices] = group_set; in ComputeDenseToSparse() 556 OutputSparseTensor<T>(ctx, output_shape, num_result_values, group_sets); in ComputeDenseToSparse() 618 std::map<std::vector<int64>, std::set<T>> group_sets; in ComputeSparseToSparse() local 667 group_sets[*group_indices] = group_set; in ComputeSparseToSparse() 679 OutputSparseTensor<T>(ctx, output_shape, num_result_values, group_sets); in ComputeSparseToSparse()
|