Lines Matching refs:n_
21 : m_(m), k_(k), n_(n), device_(device) {
26 : m_(m), k_(m), n_(m), device_(device) {
37 eigen_assert(m_ == k_ && k_ == n_);
47 eigen_assert(m_ == n_);
68 eigen_assert(m_ == k_ && k_ == n_);
83 eigen_assert(m_ == k_ && k_ == n_);
116 input_size[1] = n_;
119 output_size[0] = n_;
127 finalizeBenchmark(static_cast<int64_t>(n_) * num_iters);
133 input_size[1] = n_;
136 output_size[0] = n_;
141 C.device(device_) = B.chip(iter % n_, 1);
144 finalizeBenchmark(static_cast<int64_t>(n_) * num_iters);
148 eigen_assert(m_ == n_);
234 size_c[1] = n_;
240 broadcast[1] = n_;
245 broadcast.set(1, n_);
253 finalizeBenchmark(static_cast<int64_t>(m_) * n_ * num_iters);
257 eigen_assert(m_ == k_ && k_ == n_);
275 eigen_assert(m_ == k_ && k_ == n_);
293 eigen_assert(m_ == k_ && k_ == n_);
314 input_size[1] = n_;
317 output_size[0] = n_;
335 finalizeBenchmark(static_cast<int64_t>(k_) * n_ * num_iters);
342 input_size[1] = n_;
365 finalizeBenchmark(static_cast<int64_t>(k_) * n_ * num_iters);
372 input_size[1] = n_;
385 finalizeBenchmark(static_cast<int64_t>(k_) * n_ * num_iters);
395 sizeB[1] = n_;
398 sizeC[1] = n_;
414 finalizeBenchmark(static_cast<int64_t>(2) * m_ * n_ * k_ * num_iters);
420 input_sizes[1] = n_;
428 result_sizes[1] = n_ - kernel_y + 1;
441 (m_ - kernel_x + 1) * (n_ - kernel_y + 1) * kernel_x * kernel_y * num_iters);
447 b_ = (T *) device_.allocate(k_ * n_ * sizeof(T));
448 c_ = (T *) device_.allocate(m_ * n_ * sizeof(T));
453 device_.memset(b_, 23, k_ * n_ * sizeof(T));
454 device_.memset(c_, 31, m_ * n_ * sizeof(T));
472 TensorIndex n_;
