| /external/tensorflow/tensorflow/contrib/factorization/python/kernel_tests/ |
| D | masked_matmul_benchmark.py | 40 def _make_sparse_mask(self, mask_shape, nnz, sort=False): argument
62 def _run_graph(self, a_shape, b_shape, nnz, num_iters, sort=False, argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_tensor_dense_matmul_op_gpu.cu.cc | 31 __global__ void SparseTensorDenseMatMulKernel(int nnz, int m, int b_rows, in SparseTensorDenseMatMulKernel()
73 int nnz = a_values.size(); in Compute() local
|
| D | sparse_conditional_accumulator.h | 164 const int64 nnz = grad_idx->dim_size(0); in AllocateAndAssignToAccumGradFunction() local
323 const int64 nnz = count_element_->size(); in DivideAccumGradByCounter() local
387 const int64 nnz = grad_idx_tensor->dim_size(0); in GetAndValidateTensorInputForApplyGrad() local
424 const int64 nnz = accum_idx_vec_->size(); in ReturnIdxTensor() local
|
| D | sparse_tensor_dense_matmul_op.cc | 68 const int64 nnz = a_indices->shape().dim_size(0); in Compute() local
251 const std::size_t nnz = a_values.size(); in Compute() local
|
| D | sparse_dense_binary_op_shared.cc | 106 const int64 nnz = indices_t->dim_size(0); in Compute() local
|
| D | sparse_softmax_op.cc | 70 const int nnz = static_cast<int>(indices_t->dim_size(0)); in Compute() local
|
| D | reshape_util.cc | 54 const int64 nnz = input_indices_in.shape().dim_size(0); in Reshape() local
|
| D | sparse_tensor_dense_matmul_op_test.cc | 42 static Graph* SparseTensorDenseMatmul(int nnz, int m, int k, int n, in SparseTensorDenseMatmul()
|
| D | sparse_reduce_op.cc | 274 int64 nnz = 0; in Compute() local
|
| /external/eigen/Eigen/src/SparseCore/ |
| D | ConservativeSparseSparseProduct.h | 51 Index nnz = 0; variable
|
| D | SparseBlock.h | 46 Index nnz = 0; in nonZeros() local
|
| /external/eigen/Eigen/src/OrderingMethods/ |
| D | Ordering.h | 134 StorageIndex nnz = StorageIndex(mat.nonZeros()); in operator() local
|
| D | Eigen_Colamd.h | 257 inline IndexType colamd_recommended ( IndexType nnz, IndexType n_row, IndexType n_col) in colamd_recommended()
327 IndexType nnz ; /* nonzeros in A */ in colamd() local
|
| /external/eigen/Eigen/src/SuperLUSupport/ |
| D | SuperLUSupport.h | 139 union {int nnz;int lda;}; member
|
| /external/eigen/bench/ |
| D | sparse_setter.cpp | 302 const int nnz, in coo_tocsr()
384 I nnz = 0; in csr_sum_duplicates() local
|
| /external/libavc/encoder/ |
| D | ih264e_cavlc.c | 1233 UWORD32 *nnz; in ih264e_write_pslice_mb_cavlc() local
1544 UWORD32 *nnz; in ih264e_write_bslice_mb_cavlc() local
|
| /external/eigen/Eigen/src/IterativeLinearSolvers/ |
| D | IncompleteCholesky.h | 217 Index nnz = m_L.nonZeros(); in factorize() local
|
| /external/eigen/Eigen/src/SparseLU/ |
| D | SparseLU.h | 537 Index nnz = m_mat.nonZeros(); in factorize() local
|
| /external/eigen/test/ |
| D | sparse_basic.cpp | 73 Index nnz = internal::random<int>(1,int(rows)/2); in sparse_basic() local
|