49 lines
1.3 KiB
C++
49 lines
1.3 KiB
C++
// Copyright (C) 2018-2025 Intel Corporation
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
#pragma once
|
|
|
|
#include <cstddef>
|
|
#include <ostream>
|
|
#include <vector>
|
|
|
|
#include "openvino/core/attribute_adapter.hpp"
|
|
#include "openvino/core/core_visibility.hpp"
|
|
#include "openvino/core/rtti.hpp"
|
|
|
|
namespace ov {
|
|
/// \brief Strides for a tensor.
|
|
class Strides : public std::vector<size_t> {
|
|
public:
|
|
OPENVINO_API Strides();
|
|
|
|
OPENVINO_API Strides(const std::initializer_list<size_t>& axis_strides);
|
|
|
|
OPENVINO_API Strides(const std::vector<size_t>& axis_strides);
|
|
|
|
OPENVINO_API Strides(const Strides& axis_strides);
|
|
|
|
OPENVINO_API explicit Strides(size_t n, size_t initial_value = 0);
|
|
|
|
template <class InputIterator>
|
|
Strides(InputIterator first, InputIterator last) : std::vector<size_t>(first, last) {}
|
|
|
|
OPENVINO_API Strides& operator=(const Strides& v);
|
|
|
|
OPENVINO_API Strides& operator=(Strides&& v) noexcept;
|
|
};
|
|
|
|
OPENVINO_API
|
|
std::ostream& operator<<(std::ostream& s, const Strides& strides);
|
|
|
|
template <>
|
|
class OPENVINO_API AttributeAdapter<Strides> : public IndirectVectorValueAccessor<Strides, std::vector<int64_t>> {
|
|
public:
|
|
AttributeAdapter(Strides& value) : IndirectVectorValueAccessor<Strides, std::vector<int64_t>>(value) {}
|
|
~AttributeAdapter() override;
|
|
OPENVINO_RTTI("AttributeAdapter<Strides>");
|
|
};
|
|
|
|
} // namespace ov
|