Method RunCPU

dali/benchmark/slice_kernel_bench.cc:43–76

  }

  void RunCPU(benchmark::State& st) {
    int H = st.range(0);
    int W = st.range(1);
    int C = st.range(2);
    int anchor_h = st.range(3);
    int anchor_w = st.range(4);
    int anchor_c = st.range(5);
    int crop_h = st.range(6);
    int crop_w = st.range(7);
    int crop_c = st.range(8);

    TensorShape<Dims> in_shape{H, W, C};
    TensorShape<Dims> anchor{anchor_h, anchor_w, anchor_c};
    TensorShape<Dims> out_shape{crop_h, crop_w, crop_c};
    Setup(in_shape, out_shape);

    using Kernel = kernels::SliceCPU<OutputType, InputType, Dims>;
    Kernel kernel;

    kernels::SliceArgs<OutputType, Dims> args;
    args.anchor = anchor;
    args.shape = out_shape;

    auto out_tv = out_data.cpu()[0];
    auto in_tv = test_data.cpu()[0];

    for (auto _ : st) {
      kernels::KernelContext ctx;
      auto kernel_req = kernel.Setup(ctx, in_tv, args);
      kernel.Run(ctx, out_tv, in_tv, args);
      st.counters["FPS"] = benchmark::Counter(st.iterations() + 1,
        benchmark::Counter::kIsRate);
    }
  }
};

static void SliceKernelArgs_OnlySlice_CPU(benchmark::Benchmark *b) {