blace::ops::Interpolate2DOp Class Reference

#include <public_ops.h>

Inheritance diagram for blace::ops::Interpolate2DOp:

Public Member Functions
	Interpolate2DOp (OpP node, int height, int width, ml_core::Interpolation interpolation, bool align_corners, bool antialias)

Detailed Description

Interpolates a 2d tensor.

Constructor & Destructor Documentation

Interpolate2DOp()

blace::ops::Interpolate2DOp::Interpolate2DOp	(	OpP	node,
		int	height,
		int	width,
		ml_core::Interpolation	interpolation,
		bool	align_corners,
		bool	antialias
	)

Construct a 2d Interpolation node.

Parameters

node	The input node.
height	New height.
width	New width.
interpolation	Interpolation type.
align_corners	Align corners, see pytorch reference implementation.
antialias	Use antialiasing, see pytorch reference implementation.