Program Listing for File resize.h

↰ Return to documentation for file (include/converter/include/ops/resize.h)

#ifndef MINDSPORE_CORE_OPS_RESIZE_H_
#define MINDSPORE_CORE_OPS_RESIZE_H_
#include <memory>

#include "mindapi/base/format.h"
#include "mindapi/base/types.h"
#include "ops/base_operator.h"

namespace mindspore {
namespace ops {
constexpr auto kNameResize = "Resize";
class MIND_API Resize : public BaseOperator {
 public:
  MIND_API_BASE_MEMBER(Resize);
  Resize() : BaseOperator(kNameResize) {}

  void Init(const Format format, const ResizeMethod method, const int64_t new_height, const int64_t new_width,
            const bool preserve_aspect_ratio, const CoordinateTransformMode coordinate_transform_mode,
            const float cubic_coeff, const int64_t exclude_outside, const float extrapolation_value,
            const NearestMode nearest_mode);

  void set_format(const Format format);

  void set_method(const ResizeMethod method);

  void set_new_height(const int64_t new_height);

  void set_new_width(const int64_t new_width);

  void set_preserve_aspect_ratio(const bool preserve_aspect_ratio);

  void set_coordinate_transform_mode(const CoordinateTransformMode coordinate_transform_mode);

  void set_cubic_coeff(const float cubic_coeff);

  void set_exclude_outside(const int64_t exclude_outside);

  void set_extrapolation_value(const float extrapolation_value);

  void set_nearest_mode(const NearestMode nearest_mode);

  Format get_format() const;

  ResizeMethod get_method() const;

  int64_t get_new_height() const;

  int64_t get_new_width() const;

  bool get_preserve_aspect_ratio() const;

  CoordinateTransformMode get_coordinate_transform_mode() const;

  float get_cubic_coeff() const;

  int64_t get_exclude_outside() const;

  float get_extrapolation_value() const;

  NearestMode get_nearest_mode() const;
};
}  // namespace ops
}  // namespace mindspore

#endif  // MINDSPORE_CORE_OPS_RESIZE_H_