Function SpecDev

dali/pipeline/executor/executor2/stream_assignment_test.cc:85–90

namespace {

OpSpec SpecDev(const std::string &device) {
  return OpSpec("StreamAssignmentDummyOp")
    .AddArg("device", device)
    .AddArg("num_threads", 1)
    .AddArg("max_batch_size", 1);
}

OpSpec SpecGPU() {
  return SpecDev("gpu");