| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_utils_test.cc | 43 TTypes<int32>::ConstMatrix indices_mat(data, 8, 2); in TEST() local
46 EXPECT_TRUE(GetStartIndicesOfEachDenseRow<int32>(indices_mat, in TEST()
54 TTypes<int32>::ConstMatrix indices_mat(data, 15, 2); in TEST() local
58 GetStartIndicesOfEachDenseRow<int32>(indices_mat, in TEST()
65 TTypes<int64>::ConstMatrix indices_mat(data, 1, 2); in TEST() local
67 EXPECT_TRUE(GetStartIndicesOfEachDenseRow<int64>(indices_mat, in TEST()
74 TTypes<uint32>::ConstMatrix indices_mat(data, 2, 2); in TEST() local
76 EXPECT_TRUE(GetStartIndicesOfEachDenseRow<uint32>(indices_mat, in TEST()
83 TTypes<uint16>::ConstMatrix indices_mat(data, 4, 2); in TEST() local
86 EXPECT_TRUE(GetStartIndicesOfEachDenseRow<uint16>(indices_mat, in TEST()
[all …]
|
| D | sparse_utils.cc | 28 const typename TTypes<Tindices>::ConstMatrix& indices_mat) { in FindNextDenseRowStartIndex() argument
31 Tindices end = indices_mat.dimension(0); in FindNextDenseRowStartIndex()
35 const Tindices orig_dense_index_begin = indices_mat(begin, 0); in FindNextDenseRowStartIndex()
37 if (orig_dense_index_begin == static_cast<int64>(indices_mat(end - 1, 0))) { in FindNextDenseRowStartIndex()
43 indices_mat(begin + increment, 0) == orig_dense_index_begin) { in FindNextDenseRowStartIndex()
57 const Tindices m_dense_row_index = static_cast<Tindices>(indices_mat(m, 0)); in FindNextDenseRowStartIndex()
60 static_cast<Tindices>(indices_mat(m + 1, 0)) != in FindNextDenseRowStartIndex()
76 const typename TTypes<Tindices>::ConstMatrix& indices_mat, in GetStartIndicesOfEachDenseRow() argument
80 const Tindices num_entries_in_sparse_tensor = indices_mat.dimension(0); in GetStartIndicesOfEachDenseRow()
82 1 + indices_mat(num_entries_in_sparse_tensor - 1, 0) - indices_mat(0, 0); in GetStartIndicesOfEachDenseRow()
[all …]
|
| D | gather_nd_op_test.cc | 96 auto indices_mat = indices.matrix<Index>(); in GatherNd() local
98 indices_mat(i, 0) = rnd.Uniform(dim); in GatherNd()
99 indices_mat(i, 1) = rnd.Uniform(8); in GatherNd()
100 indices_mat(i, 2) = rnd.Uniform(16); in GatherNd()
101 indices_mat(i, 3) = rnd.Uniform(32); in GatherNd()
|
| D | sparse_utils.h | 40 const typename TTypes<Tindices>::ConstMatrix& indices_mat);
50 const typename TTypes<Tindices>::ConstMatrix& indices_mat,
|
| D | sparse_dense_binary_op_shared_test.cc | 246 auto indices_mat = indices.matrix<int64>(); in MakeSparseTensor() local
257 indices_mat(nnz_cnt, 0) = i; in MakeSparseTensor()
258 indices_mat(nnz_cnt, 1) = j; in MakeSparseTensor()
265 indices_mat(nnz_cnt, 2) = inner; in MakeSparseTensor()
|
| D | gather_nd_op.cc | 154 auto indices_mat = indices.flat_inner_dims<Index>(); in DoGatherNd() local
171 params_flat, indices_mat, out_mat); \ in DoGatherNd()
196 gtl::ArraySlice<Index>(&indices_mat(bad_i, 0), indices_nd), ", "), in DoGatherNd()
|
| D | sparse_dense_binary_op_shared.cc | 82 const auto indices_mat = indices_t->matrix<int64>(); in Compute() local
131 idx[d] = internal::SubtleMustCopy(indices_mat(i, d)); \ in Compute()
|
| /external/tensorflow/tensorflow/contrib/factorization/kernels/ |
| D | wals_solver_ops.cc | 134 ConstEigenMatrixInt64Map indices_mat(input_indices.matrix<int64>().data(), in Compute() local
153 auto get_input_index = [is_transpose, &indices_mat](int64 i) { in Compute()
154 return is_transpose ? indices_mat(1, i) : indices_mat(0, i); in Compute()
156 auto get_factor_index = [is_transpose, &indices_mat](int64 i) { in Compute()
157 return is_transpose ? indices_mat(0, i) : indices_mat(1, i); in Compute()
|
| D | masked_matmul_ops.cc | 109 ConstEigenMatInt64Map indices_mat(mask_indices.matrix<int64>().data(), in Compute() local
119 auto get_a_index = [&indices_mat, &a_dim_0](int64 i) { in Compute()
120 int64 a_index = internal::SubtleMustCopy(indices_mat(i, 0)); in Compute()
126 auto get_b_index = [&indices_mat, &b_dim_1](int64 i) { in Compute()
127 int64 b_index = internal::SubtleMustCopy(indices_mat(i, 1)); in Compute()
|