MCPcopy Create free account
hub / github.com/NVIDIA/DALI / GenerateArgs

Method GenerateArgs

dali/kernels/slice/slice_kernel_test.h:152–161  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

150 }
151
152 std::vector<SliceArgs<OutputType, Dims>> GenerateArgs(
153 const InListCPU<InputType, Dims>& input_tlv) {
154 ArgsGenerator generator;
155 std::vector<SliceArgs<OutputType, Dims>> slice_args;
156 for (int i = 0; i < NumSamples; i++) {
157 auto shape = input_tlv.tensor_shape(i);
158 slice_args.push_back(generator.Get(shape));
159 }
160 return slice_args;
161 }
162
163 virtual void Run() = 0;
164};

Callers 2

RunMethod · 0.45
RunMethod · 0.45

Calls 3

tensor_shapeMethod · 0.45
push_backMethod · 0.45
GetMethod · 0.45

Tested by

no test coverage detected