Lines Matching defs:strides
445 // cuDNN requires the strides and dims to be ordered as BDYX.
452 std::vector<int> strides(nd);
454 std::transform(strides64.cbegin(), strides64.cend(), strides.begin(),
459 parent_, handle_, elem_type, nd, dims.data(), strides.data());
622 const auto& strides64 = convolution_descriptor.strides();
631 std::vector<int> strides(convolution_descriptor.ndims());
634 std::transform(strides64.cbegin(), strides64.cend(), strides.begin(),
645 strides.data(), dilations.data(),
706 const std::vector<int64> strides64 = pooling_descriptor.strides();
713 std::vector<int> strides(nd);
714 std::transform(strides64.cbegin(), strides64.cend(), strides.begin(),
729 nd, shape.data(), padding.data(), strides.data());
1194 int strides[] = {dims[1] * dims[2], dims[2], 1};
1198 strides /*strideA*/);
1324 int strides[] = {dims[1] * dims[2], dims[2], 1};
1328 strides /*strideA*/);
1377 int strides[] = {dims[1] * dims[2], dims[2], 1};
1381 strides /*strideA*/);
