Lines Matching defs:strides
92 const std::array<int64, 3>& strides,
96 TF_RETURN_IF_ERROR(GetWindowedOutputSize(input[i], window[i], strides[i],
106 const std::array<int64, 3>& strides,
111 input[i], window[i], dilations[i], strides[i], padding_type,
413 std::vector<int32> strides;
414 TF_RETURN_IF_ERROR(c->GetAttr("strides", &strides));
416 // strides.size() should be 4 (NCHW) even if the input is 5 (NCHW_VECT_C).
417 if (strides.size() != 4) {
421 strides.size());
424 const int32 stride_rows = GetTensorDim(strides, data_format, 'H');
425 const int32 stride_cols = GetTensorDim(strides, data_format, 'W');
490 std::vector<int32> strides;
491 TF_RETURN_IF_ERROR(c->GetAttr("strides", &strides));
492 if (strides.size() != 5) {
495 strides.size());
506 stride_planes = strides[2];
507 stride_cols = strides[3];
508 stride_rows = strides[4];
510 stride_planes = strides[1];
511 stride_rows = strides[2];
512 stride_cols = strides[3];
559 std::vector<int32> strides;
560 TF_RETURN_IF_ERROR(c->GetAttr("strides", &strides));
562 if (strides.size() != 4) {
566 strides.size());
579 stride_rows = strides[2];
580 stride_cols = strides[3];
582 stride_rows = strides[1];
583 stride_cols = strides[2];
644 std::vector<int32> strides;
645 TF_RETURN_IF_ERROR(c->GetAttr("strides", &strides));
646 if (strides.size() != 4) {
649 strides.size());
660 int32 stride_rows = GetTensorDim(strides, data_format, 'H');
661 int32 stride_cols = GetTensorDim(strides, data_format, 'W');
794 std::vector<int32> strides;
795 TF_RETURN_IF_ERROR(c->GetAttr("strides", &strides));
796 if (strides.size() != 4) {
799 strides.size());
810 int32 stride_depth = GetTensorDim(strides, data_format, 'C');
811 int32 stride_rows = GetTensorDim(strides, data_format, 'H');
812 int32 stride_cols = GetTensorDim(strides, data_format, 'W');
865 std::vector<int32> strides;
870 TF_RETURN_IF_ERROR(c->GetAttr("strides", &strides));
872 // Verify shape of ksize and strides input.
895 strides.resize(strides_tensor->shape().num_elements());
897 std::copy_n(&strides_vec(0), strides.size(), strides.begin());
900 if (strides.size() != 4) {
904 strides.size());
912 int32 stride_depth = GetTensorDim(strides, data_format, 'C');
913 int32 stride_rows = GetTensorDim(strides, data_format, 'H');
914 int32 stride_cols = GetTensorDim(strides, data_format, 'W');
956 std::vector<int32> strides;
957 TF_RETURN_IF_ERROR(c->GetAttr("strides", &strides));
958 if (strides.size() != 5) {
962 strides.size());
983 stride_planes = strides[2];
984 stride_rows = strides[3];
985 stride_cols = strides[4];
990 stride_planes = strides[1];
991 stride_rows = strides[2];
992 stride_cols = strides[3];
