Thanks for sending an issue! Please fill in the following template to help quickly solve your problem.
https://gitcode.com/cann/ops-nn/blob/master/pooling/adaptive_avg_pool3d/op_graph/adaptive_avg_pool3d_proto.h
/**
/*!
#include "graph/operator_reg.h" #include "graph/operator.h"
namespace ge { /**
@brief Applies a 3D adaptive average pooling over
an input signal composed of several input planes.
@par Inputs:
One input, including:
@li x: A Tensor. Must be one of the following data types:
float16, bfloat16, float32. \n
@par Attributes:
@li output_size: A required list of 3 ints
specifying the size (D,H,W) of the output tensor. \n
@par Outputs:
@li y: A Tensor. Has the same data type as "x" \n
@li data_format: An optional string, Specify the data format of the input and
output data. With the default format "NDHWC" . \n
@par Third-party framework compatibility
Compatible with the Pytorch operator AdaptiveAvgPool3d. */ REG_OP(AdaptiveAvgPool3d) .INPUT(x, TensorType({DT_FLOAT, DT_FLOAT16, DT_BF16})) .OUTPUT(y, TensorType({DT_FLOAT, DT_FLOAT16, DT_BF16})) .REQUIRED_ATTR(output_size, ListInt) .ATTR(data_format, String, "NDHWC") .OP_END_FACTORY_REG(AdaptiveAvgPool3d)
} // namespace ge
#endif // OPS_POOLING_ADAPTIVE_AVG_POOL3D_PROTO_H_
原型库未注明输入支持的format,且参数data_format支持的类型未写全
Thanks for sending an issue! Please fill in the following template to help quickly solve your problem.
Document Link(文档链接)
https://gitcode.com/cann/ops-nn/blob/master/pooling/adaptive_avg_pool3d/op_graph/adaptive_avg_pool3d_proto.h
Issues Section(问题文档片段)
/**
*/
/*!
*/
#ifndef OPS_POOLING_ADAPTIVE_AVG_POOL3D_PROTO_H_
#define OPS_POOLING_ADAPTIVE_AVG_POOL3D_PROTO_H_
#include "graph/operator_reg.h"
#include "graph/operator.h"
namespace ge {
/**
@brief Applies a 3D adaptive average pooling over
an input signal composed of several input planes.
@par Inputs:
One input, including:
@li x: A Tensor. Must be one of the following data types:
@par Attributes:
@li output_size: A required list of 3 ints
specifying the size (D,H,W) of the output tensor. \n
@par Outputs:
@li y: A Tensor. Has the same data type as "x" \n
@li data_format: An optional string, Specify the data format of the input and
output data. With the default format "NDHWC" . \n
@par Third-party framework compatibility
Compatible with the Pytorch operator AdaptiveAvgPool3d.
*/
REG_OP(AdaptiveAvgPool3d)
.INPUT(x, TensorType({DT_FLOAT, DT_FLOAT16, DT_BF16}))
.OUTPUT(y, TensorType({DT_FLOAT, DT_FLOAT16, DT_BF16}))
.REQUIRED_ATTR(output_size, ListInt)
.ATTR(data_format, String, "NDHWC")
.OP_END_FACTORY_REG(AdaptiveAvgPool3d)
} // namespace ge
#endif // OPS_POOLING_ADAPTIVE_AVG_POOL3D_PROTO_H_
Existing Issues(存在的问题)
原型库未注明输入支持的format,且参数data_format支持的类型未写全