Home | History | Annotate | Download | only in tensors

Lines Matching refs:m_

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_);
40 device_.memcpy(c_, a_, m_ * m_ * sizeof(T));
43 finalizeBenchmark(static_cast<int64_t>(m_) * m_ * num_iters);
47 eigen_assert(m_ == n_);
50 sizes[0] = m_;
53 sizes[0] = m_ * sizeof(T) / sizeof(int);
64 finalizeBenchmark(static_cast<int64_t>(m_) * k_ * num_iters);
68 eigen_assert(m_ == k_ && k_ == n_);
70 sizes[0] = m_;
71 sizes[1] = m_;
79 finalizeBenchmark(static_cast<int64_t>(m_) * m_ * num_iters);
83 eigen_assert(m_ == k_ && k_ == n_);
85 sizes[0] = m_;
86 sizes[1] = m_;
91 const Eigen::DSizes<TensorIndex, 2> quarter_sizes(m_/2, m_/2);
93 const Eigen::DSizes<TensorIndex, 2> second_quadrant(0, m_/2);
94 const Eigen::DSizes<TensorIndex, 2> third_quadrant(m_/2, 0);
95 const Eigen::DSizes<TensorIndex, 2> fourth_quadrant(m_/2, m_/2);
110 finalizeBenchmark(static_cast<int64_t>(m_) * m_ * num_iters);
148 eigen_assert(m_ == n_);
150 size_a[0] = m_;
155 size_b[1] = m_;
167 finalizeBenchmark(static_cast<int64_t>(m_) * k_ * num_iters);
171 eigen_assert(m_ == k_);
173 size_a[0] = m_;
178 size_b[1] = m_;
195 finalizeBenchmark(static_cast<int64_t>(m_) * k_ * num_iters);
199 eigen_assert(m_ == k_);
201 size_a[0] = m_;
205 size_b[0] = m_;
224 finalizeBenchmark(static_cast<int64_t>(m_) * k_ * num_iters);
229 size_a[0] = m_;
233 size_c[0] = m_;
253 finalizeBenchmark(static_cast<int64_t>(m_) * n_ * num_iters);
257 eigen_assert(m_ == k_ && k_ == n_);
259 sizes[0] = m_;
260 sizes[1] = m_;
271 finalizeBenchmark(static_cast<int64_t>(3) * m_ * m_ * num_iters);
275 eigen_assert(m_ == k_ && k_ == n_);
277 sizes[0] = m_;
278 sizes[1] = m_;
289 finalizeBenchmark(static_cast<int64_t>(m_) * m_ * num_iters);
293 eigen_assert(m_ == k_ && k_ == n_);
295 sizes[0] = m_;
296 sizes[1] = m_;
307 finalizeBenchmark(static_cast<int64_t>(m_) * m_ * num_iters);
391 sizeA[0] = m_;
397 sizeC[0] = m_;
414 finalizeBenchmark(static_cast<int64_t>(2) * m_ * n_ * k_ * num_iters);
419 input_sizes[0] = m_;
427 result_sizes[0] = m_ - kernel_x + 1;
441 (m_ - kernel_x + 1) * (n_ - kernel_y + 1) * kernel_x * kernel_y * num_iters);
446 a_ = (T *) device_.allocate(m_ * k_ * sizeof(T));
448 c_ = (T *) device_.allocate(m_ * n_ * sizeof(T));
452 device_.memset(a_, 12, m_ * k_ * sizeof(T));
454 device_.memset(c_, 31, m_ * n_ * sizeof(T));
470 TensorIndex m_;