m53297601
/
graphengine
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

update include file 0325

Browse Source
pull/1354/head
shenwei41 4 years ago
parent e2f929b761
commit 23b471ca2b
3 changed files with 114 additions and 36 deletions
Show Stats Download Patch File Download Diff File

79
third_party/fwkacllib/inc/ops/nn_pooling_ops.h vendored
Unescape View File

@ -1444,8 +1444,7 @@ REG_OP(MaxPoolV3Grad)
*@par Inputs: *@par Inputs:
*x: A tensor of shape is 4d, format is support NHWC. *x: A tensor of shape is 4d, format is support NHWC.
*filter: A tensor of shape is 3d, the type is same with x, *filter: A tensor of shape is 3d, the type is same with x, and the c dimension is same with x. \n
and the c dimension is same with x. \n
*@par Attributes: *@par Attributes:
*@li strides: A required list of 4 ints, specifying the stride of the sliding window. The strides of the N and C dimensions are 1. *@li strides: A required list of 4 ints, specifying the stride of the sliding window. The strides of the N and C dimensions are 1.
@ -1473,6 +1472,82 @@ REG_OP(Dilation2D)
.ATTR(data_format, String, "NHWC") .ATTR(data_format, String, "NHWC")
.OP_END_FACTORY_REG(Dilation2D) .OP_END_FACTORY_REG(Dilation2D)
/*
* @brief Performs Dilation2DBackpropFilter on the input. \n
*@par Inputs:
*x: A tensor of shape is 4d, format is support NHWC
*filter: A tensor of shape is 3d the type is same with x,
*out_backprop: Has the same type and format as input "x" and the c dimension is same with x. \n
*@par Attributes
*@li stride: A required list of 4 ints, specifying the stride of the sliding window. The strides of the N and C
dimension are 1
*@li rates: A required list of 4 ints, the rates of the N and C dimensions are 1
*@li padding_mode: A optional string. Defaults to "SAME", it support SAME and VALID
*@li pads: A optional list of 4 ints. \n
*@par Outputs:
*y: The output tensor. Has the same type and format as input "filter" . \n
*@par Third-party framework compatibility
* Compatible with the TensorFlow operator Dilation2D
*/
REG_OP(Dilation2DBackpropFilter)
.INPUT(x, TensorType({DT_FLOAT16, DT_FLOAT, DT_DOUBLE, DT_INT32, DT_INT64, DT_UINT8, DT_INT16, DT_INT8, DT_UINT16}))
.INPUT(filter,
TensorType({DT_FLOAT16, DT_FLOAT, DT_DOUBLE, DT_INT32, DT_INT64, DT_UINT8, DT_INT16, DT_INT8, DT_UINT16}))
.INPUT(out_backprop,
TensorType({DT_FLOAT16, DT_FLOAT, DT_DOUBLE, DT_INT32, DT_INT64, DT_UINT8, DT_INT16, DT_INT8, DT_UINT16}))
.OUTPUT(y,
TensorType({DT_FLOAT16, DT_FLOAT, DT_DOUBLE, DT_INT32, DT_INT64, DT_UINT8, DT_INT16, DT_INT8, DT_UINT16}))
.REQUIRED_ATTR(strides, ListInt)
.REQUIRED_ATTR(rates, ListInt)
.ATTR(padding_mode, String, "SAME")
.ATTR(pads, ListInt, {0, 0, 0, 0})
.ATTR(ceil_mode, Bool, false)
.ATTR(data_format, String, "NHWC")
.OP_END_FACTORY_REG(Dilation2DBackpropFilter)
/*
* @brief Performs Dilation2DBackpropInput on the input. \n
*@par Inputs:
*x: A tensor of shape is 4d, format is support NHWC
*filter: A tensor of shape is 3d the type is same with x,
*out_backprop: Has the same type and format as input "x" and the c dimension is same with x. \n
*@par Attributes
*@li stride: A required list of 4 ints, specifying the stride of the sliding window. The strides of the N and C
dimension are 1
*@li rates: A required list of 4 ints, the rates of the N and C dimensions are 1
*@li padding_mode: A optional string. Defaults to "SAME", it support SAME and VALID
*@li pads: A optional list of 4 ints. \n
*@par Outputs:
*y: The output tensor. Has the same type and format as input "filter" . \n
*@par Third-party framework compatibility
* Compatible with the TensorFlow operator Dilation2D
*/
REG_OP(Dilation2DBackpropInput)
.INPUT(x, TensorType({DT_FLOAT16, DT_FLOAT, DT_DOUBLE, DT_INT32, DT_INT64, DT_UINT8, DT_INT16, DT_INT8, DT_UINT16}))
.INPUT(filter,
TensorType({DT_FLOAT16, DT_FLOAT, DT_DOUBLE, DT_INT32, DT_INT64, DT_UINT8, DT_INT16, DT_INT8, DT_UINT16}))
.INPUT(out_backprop,
TensorType({DT_FLOAT16, DT_FLOAT, DT_DOUBLE, DT_INT32, DT_INT64, DT_UINT8, DT_INT16, DT_INT8, DT_UINT16}))
.OUTPUT(y,
TensorType({DT_FLOAT16, DT_FLOAT, DT_DOUBLE, DT_INT32, DT_INT64, DT_UINT8, DT_INT16, DT_INT8, DT_UINT16}))
.REQUIRED_ATTR(strides, ListInt)
.REQUIRED_ATTR(rates, ListInt)
.ATTR(padding_mode, String, "SAME")
.ATTR(pads, ListInt, {0, 0, 0, 0})
.ATTR(ceil_mode, Bool, false)
.ATTR(data_format, String, "NHWC")
.OP_END_FACTORY_REG(Dilation2DBackpropInput)
/** /**
* @brief Applies a 2D adaptive average pooling over * @brief Applies a 2D adaptive average pooling over
* an input signal composed of several input planes. \n * an input signal composed of several input planes. \n

1
third_party/fwkacllib/inc/toolchain/prof_callback.h vendored
Unescape View File

@ -74,6 +74,7 @@ enum MsprofReporterCallbackType {
MSPROF_REPORTER_REPORT = 0, // report data MSPROF_REPORTER_REPORT = 0, // report data
MSPROF_REPORTER_INIT, // init reporter MSPROF_REPORTER_INIT, // init reporter
MSPROF_REPORTER_UNINIT, // uninit reporter MSPROF_REPORTER_UNINIT, // uninit reporter
MSPROF_REPORTER_DATA_MAX_LEN, // data max length for calling report callback
}; };
/** /**

70
third_party/fwkacllib/inc/toolchain/prof_reporter.h vendored
Unescape View File

@ -41,42 +41,44 @@ namespace Engine {
* the Reporter class .used to send data to profiling * the Reporter class .used to send data to profiling
*/ */
class MSVP_PROF_API Reporter { class MSVP_PROF_API Reporter {
public: public:
virtual ~Reporter() {} virtual ~Reporter() {}
public: public:
/** /**
* @ingroup reporter * @ingroup reporter
* @name : Report * @name : Report
* @brief : API of libmsprof, report data to libmsprof, it's a non-blocking function \n * @brief : API of libmsprof, report data to libmsprof, it's a non-blocking function \n
The data will be firstly appended to cache, if the cache is full, data will be ignored The data will be firstly appended to cache, if the cache is full, data will be ignored
* @param data [IN] const ReporterData * the data send to libmsporf * @param data [IN] const ReporterData * the data send to libmsporf
* @retval PROFILING_SUCCESS 0 (success) * @retval PROFILING_SUCCESS 0 (success)
* @retval PROFILING_FAILED -1 (failed) * @retval PROFILING_FAILED -1 (failed)
* *
* @par depend: * @par depend:
* @li libmsprof * @li libmsprof
* @li prof_reporter.h * @li prof_reporter.h
* @since c60 * @since c60
* @see Flush * @see Flush
*/ */
virtual int Report(const ReporterData *data) = 0; virtual int Report(const ReporterData *data) = 0;
/** /**
* @ingroup reporter * @ingroup reporter
* @name : Flush * @name : Flush
* @brief : API of libmsprof, notify libmsprof send data over, it's a blocking function \n * @brief : API of libmsprof, notify libmsprof send data over, it's a blocking function \n
The all datas of cache will be write to file or send to host The all datas of cache will be write to file or send to host
* @retval PROFILING_SUCCESS 0 (success) * @retval PROFILING_SUCCESS 0 (success)
* @retval PROFILING_FAILED -1 (failed) * @retval PROFILING_FAILED -1 (failed)
* *
* @par depend: * @par depend:
* @li libmsprof * @li libmsprof
* @li prof_reporter.h * @li prof_reporter.h
* @since c60 * @since c60
* @see ProfMgrStop * @see ProfMgrStop
*/ */
virtual int Flush() = 0; virtual int Flush() = 0;
virtual uint32_t GetReportDataMaxLen() = 0;
}; };
} // namespace Engine } // namespace Engine

Write Preview
Loading…
Cancel
Save