Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dset_kernels.cc441 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()