Searched refs:c_values_t (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mul_op.cc | 133 Tensor c_values_t; in Compute() local
135 DataTypeToEnum<T>::value, TensorShape({total_nnz}), &c_values_t)); in Compute()
138 a.row_pointers(), a.col_indices(), c_values_t, c)); in Compute()
141 auto c_values = c_values_t.flat<T>(); in Compute()
|
| D | add_op.cc | 130 Tensor c_values_t; in operator ()() local
137 DataTypeToEnum<T>::value, TensorShape({total_nnz}), &c_values_t)); in operator ()()
140 c_col_ind_t, c_values_t, c)); in operator ()()
|
| D | kernels_gpu.cu.cc | 303 Tensor c_values_t; in CSRSparseMatrixBatchMulVecImpl() local
305 TensorShape({total_nnz}), &c_values_t)); in CSRSparseMatrixBatchMulVecImpl()
308 a.row_pointers(), a.col_indices(), c_values_t, c)); in CSRSparseMatrixBatchMulVecImpl()
311 auto c_values = c_values_t.flat<T>(); in CSRSparseMatrixBatchMulVecImpl()
|
| D | sparse_mat_mul_op.cc | 442 Tensor c_values_t; in Compute() local
450 TensorShape({total_nnz}), &c_values_t)); in Compute()
454 c_row_ptr_t, c_col_ind_t, c_values_t, &c)); in Compute()
|