#include <tensor_benchmarks.h>
Public Member Functions | |
void | algebraicFunc (int num_iters) |
BenchmarkSuite (const Device &device, size_t m, size_t k, size_t n) | |
BenchmarkSuite (const Device &device, size_t m) | |
BenchmarkSuite (const Device &device, size_t m, size_t k) | |
void | broadcasting (int num_iters) |
void | coeffWiseOp (int num_iters) |
void | colChip (int num_iters) |
void | colReduction (int num_iters) |
void | contraction (int num_iters) |
void | contractionRowMajor (int num_iters) |
void | contractionRowMajorABT (int num_iters) |
void | contractionRowMajorAT (int num_iters) |
void | contractionRowMajorBT (int num_iters) |
void | convolution (int num_iters, int kernel_x, int kernel_y) |
void | fullReduction (int num_iters) |
void | memcpy (int num_iters) |
void | padding (int num_iters) |
void | random (int num_iters) |
void | rowChip (int num_iters) |
void | rowReduction (int num_iters) |
void | shuffling (int num_iters) |
void | slicing (int num_iters) |
void | striding (int num_iters) |
void | transcendentalFunc (int num_iters) |
void | typeCasting (int num_iters) |
~BenchmarkSuite () | |
Private Member Functions | |
template<int Layout> | |
void | contraction (int num_iters, bool trans_a, bool trans_b) |
void | finalizeBenchmark (int64_t num_items) |
void | initialize () |
Private Attributes | |
T * | a_ |
T * | b_ |
T * | c_ |
Device | device_ |
TensorIndex | k_ |
TensorIndex | m_ |
TensorIndex | n_ |
Definition at line 18 of file tensor_benchmarks.h.
|
inline |
Definition at line 20 of file tensor_benchmarks.h.
|
inline |
Definition at line 25 of file tensor_benchmarks.h.
|
inline |
Definition at line 30 of file tensor_benchmarks.h.
|
inline |
Definition at line 35 of file tensor_benchmarks.h.
|
inline |
Definition at line 334 of file tensor_benchmarks.h.
|
inline |
Definition at line 278 of file tensor_benchmarks.h.
|
inline |
Definition at line 312 of file tensor_benchmarks.h.
|
inline |
Definition at line 163 of file tensor_benchmarks.h.
|
inline |
Definition at line 412 of file tensor_benchmarks.h.
|
inline |
Definition at line 472 of file tensor_benchmarks.h.
|
inlineprivate |
Definition at line 526 of file tensor_benchmarks.h.
|
inline |
Definition at line 476 of file tensor_benchmarks.h.
|
inline |
Definition at line 488 of file tensor_benchmarks.h.
|
inline |
Definition at line 480 of file tensor_benchmarks.h.
|
inline |
Definition at line 484 of file tensor_benchmarks.h.
|
inline |
Definition at line 492 of file tensor_benchmarks.h.
|
inlineprivate |
Definition at line 573 of file tensor_benchmarks.h.
|
inline |
Definition at line 446 of file tensor_benchmarks.h.
|
inlineprivate |
Definition at line 560 of file tensor_benchmarks.h.
|
inline |
Definition at line 41 of file tensor_benchmarks.h.
|
inline |
Definition at line 211 of file tensor_benchmarks.h.
|
inline |
Definition at line 81 of file tensor_benchmarks.h.
|
inline |
Definition at line 142 of file tensor_benchmarks.h.
|
inline |
Definition at line 380 of file tensor_benchmarks.h.
|
inline |
Definition at line 184 of file tensor_benchmarks.h.
|
inline |
Definition at line 100 of file tensor_benchmarks.h.
|
inline |
Definition at line 243 of file tensor_benchmarks.h.
|
inline |
Definition at line 357 of file tensor_benchmarks.h.
|
inline |
Definition at line 56 of file tensor_benchmarks.h.
|
private |
Definition at line 592 of file tensor_benchmarks.h.
|
private |
Definition at line 593 of file tensor_benchmarks.h.
|
private |
Definition at line 594 of file tensor_benchmarks.h.
|
private |
Definition at line 595 of file tensor_benchmarks.h.
|
private |
Definition at line 590 of file tensor_benchmarks.h.
|
private |
Definition at line 589 of file tensor_benchmarks.h.
|
private |
Definition at line 591 of file tensor_benchmarks.h.