Searched refs:batch_pointers (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | sparse_matrix.h | 147 const Tensor& batch_pointers, // on host in CreateCSRSparseMatrix() argument
152 *matrix = CSRSparseMatrix(dtype, dense_shape, batch_pointers, row_pointers, in CreateCSRSparseMatrix()
305 inline Tensor& batch_pointers() { in batch_pointers() function
310 inline const Tensor& batch_pointers() const { in batch_pointers() function
344 Tensor batch_pointers(p.tensors_[1]); in Decode()
350 Status s = ValidateTypesAndShapes(dtype, dense_shape, batch_pointers, in Decode()
357 batch_pointers_ = std::move(batch_pointers); in Decode()
395 from.batch_pointers(), // Always on host. in DeviceCopy()
401 const Tensor& batch_pointers, const Tensor& row_pointers, in CSRSparseMatrix() argument
405 batch_pointers_(batch_pointers),
[all …]
|
| D | conj_op.cc | 56 DataTypeToEnum<T>::value, a.dense_shape(), a.batch_pointers(), in operator ()()
78 DataTypeToEnum<T>::value, a.dense_shape(), a.batch_pointers(), \
|
| D | softmax_op.cc | 72 logits_matrix->batch_pointers(), logits_matrix->row_pointers(), in Compute()
179 softmax_matrix->batch_pointers(), in Compute()
|
| D | kernels_gpu.cu.cc | 294 DataTypeToEnum<T>::value, a.dense_shape(), a.batch_pointers(), in CSRSparseMatrixBatchMulVecImpl()
301 auto host_batch_ptr = a.batch_pointers().vec<int>(); in CSRSparseMatrixBatchMulVecImpl()
423 auto host_batch_ptr = logits.batch_pointers().vec<int32>(); in CSRSparseMatrixSoftmaxGPUImpl()
612 auto softmax_host_batch_ptr = softmax.batch_pointers().vec<int32>(); in CSRSparseMatrixSoftmaxGradGPUImpl()
616 auto grad_softmax_host_batch_ptr = grad_softmax.batch_pointers().vec<int32>(); in CSRSparseMatrixSoftmaxGradGPUImpl()
|
| D | csr_sparse_matrix_to_sparse_tensor_op.cc | 96 auto batch_ptrs = csr_sparse_matrix->batch_pointers().vec<int32>(); in Compute()
169 auto batch_ptrs = csr_sparse_matrix->batch_pointers().vec<int32>(); in Compute()
|
| D | csr_sparse_matrix_to_dense_op.cc | 76 auto batch_ptrs = csr_sparse_matrix->batch_pointers().vec<int32>(); in Compute()
163 auto batch_ptrs = csr_sparse_matrix->batch_pointers().vec<int32>(); in Compute()
|
| D | sparse_matrix_components_op.cc | 70 auto batch_ptrs = csr_sparse_matrix->batch_pointers().vec<int32>(); in Compute()
|
| D | mul_op.cc | 135 DataTypeToEnum<T>::value, a.dense_shape(), a.batch_pointers(), in Compute()
|
| D | transpose_op.cc | 167 Tensor batch_ptr_t = input_matrix.batch_pointers(); in operator ()()
|