Home
last modified time | relevance | path

Searched refs:DIM1 (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dbroadcast_to_op_test.cc47 #define BM_BroadcastTo_InnerDim(DIM0, DIM1, type) \ argument
48 static void BM_BroadcastTo_Inner##_##type##_##DIM0##_##DIM1(int iters) { \
50 testing::ItemsProcessed(static_cast<int64>(iters) * DIM0 * DIM1); \
51 test::Benchmark(#type, BroadcastTo(DIM0, DIM1, \
57 BENCHMARK(BM_BroadcastTo_Inner##_##type##_##DIM0##_##DIM1);
59 #define BM_BroadcastTo_OuterDim(DIM0, DIM1, type) \ argument
60 static void BM_BroadcastTo_Outer##_##type##_##DIM0##_##DIM1(int iters) { \
62 testing::ItemsProcessed(static_cast<int64>(iters) * DIM0 * DIM1); \
63 test::Benchmark(#type, BroadcastTo(DIM0, DIM1, \
69 BENCHMARK(BM_BroadcastTo_Outer##_##type##_##DIM0##_##DIM1);
/external/eigen/bench/tensors/
Dtensor_benchmarks_fp16_gpu.cu59 #define BM_FuncWithKernelDimsGPU(FUNC, DIM1, DIM2) \ argument
60 static void BM_##FUNC##_##DIM1##x##DIM2(int iters, int N) { \
66 suite.FUNC(iters, DIM1, DIM2); \
68 BENCHMARK_RANGE(BM_##FUNC##_##DIM1##x##DIM2, 128, 5000);
Dtensor_benchmarks_gpu.cu59 #define BM_FuncWithKernelDimsGPU(FUNC, DIM1, DIM2) \ argument
60 static void BM_##FUNC##_##DIM1##x##DIM2(int iters, int N) { \
66 suite.FUNC(iters, DIM1, DIM2); \
68 BENCHMARK_RANGE(BM_##FUNC##_##DIM1##x##DIM2, 128, 5000);
Dtensor_benchmarks_cpu.cc137 #define BM_FuncWithKernelDimsCPU(FUNC, DIM1, DIM2, THREADS) \ argument
138 static void BM_##FUNC##_##DIM1##x##DIM2##_##THREADS##T(int iters, int N) { \
142 suite.FUNC(iters, DIM1, DIM2); \
144 BENCHMARK_RANGE(BM_##FUNC##_##DIM1##x##DIM2##_##THREADS##T, 128, 5000);