HomeSort by relevance Sort by last modified time
    Searched refs:NNZ (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
sparse_tensor_dense_matmul_op_test.cc 42 static Graph* SparseTensorDenseMatmul(int nnz, int m, int k, int n,
45 Tensor a_values(DT_FLOAT, TensorShape({nnz}));
46 Tensor a_indices(DT_INT64, TensorShape({nnz, 2}));
57 for (int32 i = 0; i < nnz; ++i) {
71 #define BM_SparseTensorDenseMatmulDev(NNZ, M, K, N, TA, TB, DEVICE) \
73 BM_SparseTensorDenseMatmul##_##NNZ##_##M##_##K##_##N##_##TA##_##TB##_##DEVICE( \
75 int64 items_per_iter = (static_cast<int64>(NNZ) * (TB ? K : N)); \
79 test::Benchmark(#DEVICE, SparseTensorDenseMatmul(NNZ, M, K, N, TA, TB)) \
83 BM_SparseTensorDenseMatmul##_##NNZ##_##M##_##K##_##N##_##TA##_##TB##_##DEVICE);
85 #define BM_SparseTensorDenseMatmul(NNZ, M, K, N, TA, TB)
    [all...]
  /external/eigen/unsupported/Eigen/src/SparseExtra/
MarketIO.h 149 Index M(-1), N(-1), NNZ(-1);
162 line >> M >> N >> NNZ;
163 if(M > 0 && N > 0 && NNZ > 0)
166 //std::cout << "sizes: " << M << "," << N << "," << NNZ << "\n";
168 mat.reserve(NNZ);
185 if(count!=NNZ)
186 std::cerr << count << "!=" << NNZ << "\n";

Completed in 250 milliseconds