Searched refs:UnalignedVec (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | sparse_matrix.h | 219 inline TTypes<int32>::UnalignedVec row_pointers_vec(int batch) { in row_pointers_vec()
224 return TTypes<int32>::UnalignedVec(row_pointers_vec_->data() + offset, in row_pointers_vec()
237 inline TTypes<int32>::UnalignedVec col_indices_vec(int batch) { in col_indices_vec()
242 return TTypes<int32>::UnalignedVec(col_indices_vec_->data() + offset, in col_indices_vec()
256 inline typename TTypes<T>::UnalignedVec values_vec(int batch) { in values_vec()
261 return typename TTypes<T>::UnalignedVec(&(values().vec<T>()(offset)), in values_vec()
626 TTypes<int32>::UnalignedVec row_ptr;
627 TTypes<int32>::UnalignedVec col_ind;
628 typename TTypes<T>::UnalignedVec values;
|
| D | kernels.h | 103 TTypes<int32>::UnalignedVec coo_row_ind,
104 TTypes<int32>::UnalignedVec csr_row_ptr);
140 TTypes<int32>::UnalignedVec coo_row_ind);
176 TTypes<int32>::UnalignedVec c_row_ptr,
|
| D | sparse_tensor_to_csr_sparse_matrix_op.cc | 254 TTypes<int32>::UnalignedVec(&coo_row_ind(batch_ptr(i)), nnz_i); in ComputeAsync()
255 auto csr_row_ptr_i = TTypes<int32>::UnalignedVec( in ComputeAsync()
309 TTypes<int>::UnalignedVec coo_row_ind, in operator ()()
310 TTypes<int>::UnalignedVec csr_row_ptr) { in operator ()()
|
| D | csr_sparse_matrix_to_sparse_tensor_op.cc | 186 const TTypes<int32>::UnalignedVec coo_row_ind_i( in Compute()
238 TTypes<const int>::UnalignedVec csr_row_ptr,
239 TTypes<int>::UnalignedVec coo_row_ind);
|
| D | csr_sparse_matrix_to_dense_op.cc | 180 const TTypes<int32>::UnalignedVec coo_row_ind_i( in Compute()
252 TTypes<const int>::UnalignedVec csr_row_ptr,
253 TTypes<int>::UnalignedVec coo_row_ind);
|
| D | dense_to_csr_sparse_matrix_op.cc | 321 TTypes<int32>::UnalignedVec(&coo_row_ind(batch_ptr(i)), nnz_i); in ComputeAsync()
322 auto csr_row_ptr_i = TTypes<int32>::UnalignedVec( in ComputeAsync()
387 TTypes<int>::UnalignedVec coo_row_ind, in operator ()()
388 TTypes<int>::UnalignedVec csr_row_ptr) { in operator ()()
|
| D | add_op.cc | 141 TTypes<int32>::UnalignedVec c_row_ptr_i(&c_row_ptr(i * (rows + 1)), in operator ()()
320 TTypes<int32>::UnalignedVec c_row_ptr, in GetOutputStructure()
|
| D | sparse_mat_mul_op.cc | 462 TTypes<int32>::UnalignedVec c_row_ptr_i(&c_row_ptr(i * (rows + 1)), in Compute()
632 TTypes<int32>::UnalignedVec c_row_ptr, in CSRSparseSparseMatrixMatMul()
|
| D | kernels_gpu.cu.cc | 126 OpKernelContext* c, TTypes<const int>::UnalignedVec csr_row_ptr, in operator ()()
127 TTypes<int>::UnalignedVec coo_row_ind) { in operator ()()
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | group_iterator.h | 43 typename TTypes<T>::UnalignedVec values() const;
145 typename TTypes<T>::UnalignedVec Group::values() const { in values()
146 return typename TTypes<T>::UnalignedVec(&(iter_->vals_.vec<T>()(loc_)), in values()
|
| D | README.md | 114 TTypes<tstring>::UnalignedVec g_v = g.values();
|
| D | sparse_tensor_test.cc | 471 std::vector<TTypes<int32>::UnalignedVec> grouped_values; in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_reduce_op.cc | 135 …n(OpKernelContext *ctx, typename TTypes<T>::Scalar &s, const typename TTypes<T>::UnalignedVec &v) { in Run()
145 …n(OpKernelContext *ctx, typename TTypes<T>::Scalar &s, const typename TTypes<T>::UnalignedVec &v) { in Run()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_types.h | 84 UnalignedVec; typedef
|