38 lines
1.3 KiB
C++
38 lines
1.3 KiB
C++
|
// Copyright (C) 2018-2025 Intel Corporation
|
||
|
|
// SPDX-License-Identifier: Apache-2.0
|
||
|
|
//
|
||
|
|
|
||
|
|
#pragma once
|
||
|
|
|
||
|
|
#include "openvino/op/op.hpp"
|
||
|
|
#include "openvino/op/util/reduction_base.hpp"
|
||
|
|
|
||
|
|
namespace ov {
|
||
|
|
namespace op {
|
||
|
|
namespace util {
|
||
|
|
/// \brief Abstract base class for logical reduction operations, i.e., operations where
|
||
|
|
/// chosen axes of the input tensors are eliminated (reduced out) by repeated
|
||
|
|
/// application of a particular binary logical operation.
|
||
|
|
class OPENVINO_API LogicalReduction : public ReductionBase {
|
||
|
|
protected:
|
||
|
|
/// \brief Constructs a logical reduction operation.
|
||
|
|
LogicalReduction();
|
||
|
|
/// \brief Constructs a logical reduction operation.
|
||
|
|
///
|
||
|
|
/// \param arg Output that produces the first input tensor.
|
||
|
|
/// \param reduction_axes The axis positions (0-based) to be eliminated.
|
||
|
|
LogicalReduction(const Output<Node>& arg, const AxisSet& reduction_axes);
|
||
|
|
/// \brief Constructs a 'dynamic' logical reduction operation.
|
||
|
|
///
|
||
|
|
/// \param arg Node that produces the first input tensor.
|
||
|
|
/// \param reduction_axes The axis positions (0-based) to be eliminated.
|
||
|
|
LogicalReduction(const Output<Node>& arg, const Output<Node>& reduction_axes);
|
||
|
|
|
||
|
|
public:
|
||
|
|
OPENVINO_OP("LogicalReduction", "util", ReductionBase);
|
||
|
|
void validate_and_infer_types() override;
|
||
|
|
};
|
||
|
|
} // namespace util
|
||
|
|
} // namespace op
|
||
|
|
} // namespace ov
|