OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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