Home
last modified time | relevance | path

Searched defs:indices_mat (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_utils_test.cc46 TTypes<int32>::ConstMatrix indices_mat(data, 8, 2); in TEST() local
57 TTypes<int32>::ConstMatrix indices_mat(data, 15, 2); in TEST() local
68 TTypes<int64_t>::ConstMatrix indices_mat(data, 1, 2); in TEST() local
77 TTypes<uint32>::ConstMatrix indices_mat(data, 2, 2); in TEST() local
86 TTypes<uint16>::ConstMatrix indices_mat(data, 4, 2); in TEST() local
96 TTypes<uint64>::ConstMatrix indices_mat(data, 4, 2); in TEST() local
148 TTypes<int32>::ConstMatrix indices_mat(data, 8, 2); in TEST() local
157 TTypes<int64_t>::ConstMatrix indices_mat(data, 8, 2); in TEST() local
166 TTypes<int32>::ConstMatrix indices_mat(data, 6, 2); in TEST() local
175 TTypes<uint16>::ConstMatrix indices_mat(data, 6, 2); in TEST() local
[all …]
H A Dsparse_utils.cc32 const typename TTypes<Tindices>::ConstMatrix& indices_mat) { in FindNextDenseRowStartIndex()
80 const typename TTypes<Tindices>::ConstMatrix& indices_mat, in GetStartIndicesOfEachDenseRow()
202 const auto indices_mat = indices.flat_inner_dims<Tindices>(); in ValidateSparseTensorIndicesUnordered() local
226 const auto indices_mat = indices.flat_inner_dims<Tindices>(); in ValidateSparseTensorIndicesOrdered() local
H A Dgather_nd_op.h121 auto indices_mat = indices.flat_inner_dims<Index>(); in DoGatherNd() local
H A Dsparse_dense_binary_op_shared.cc100 const auto indices_mat = indices_t->matrix<int64_t>(); in Compute() local
H A Dgather_nd_op_test.cc122 auto indices_mat = indices.matrix<Index>(); in GatherNd() local
H A Dsparse_dense_binary_op_shared_test.cc247 auto indices_mat = indices.matrix<int64_t>(); in MakeSparseTensor() local
H A Dbincount_op.cc423 const auto indices_mat = indices.matrix<int64_t>(); in Compute() local