We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 127015f commit ecf8432Copy full SHA for ecf8432
dali/pipeline/operator/builtin/make_contiguous.cu
@@ -21,7 +21,7 @@ namespace dali {
21
22
void MakeContiguousMixed::RunImpl(Workspace &ws) {
23
const auto& input = ws.Input<CPUBackend>(0);
24
- int sample_dim = input[0].shape().sample_dim();
+ int sample_dim = input.shape().sample_dim();
25
size_t batch_size = input.num_samples();
26
DALIDataType type = input.type();
27
size_t type_size = input.type_info().size();
0 commit comments