Author | SHA1 | Message | Date |
---|---|---|---|
|
8ac63cd76d
|
!2136 upgrade Ascend software package 3 Nov 22
Merge pull request !2136 from yanghaoran/update_with_ascend_newest |
2 years ago |
|
a693b462bb | upgrade Ascend software package 3 Nov 22 | 2 years ago |
@@ -25,9 +25,9 @@ | |||
extern "C" { | |||
#endif | |||
// Current version is 1.3.0 | |||
// Current version is 1.4.0 | |||
#define ACL_MAJOR_VERSION 1 | |||
#define ACL_MINOR_VERSION 3 | |||
#define ACL_MINOR_VERSION 4 | |||
#define ACL_PATCH_VERSION 0 | |||
/** | |||
@@ -98,6 +98,12 @@ typedef enum aclrtGroupAttr { | |||
ACL_GROUP_GROUPID_INT | |||
} aclrtGroupAttr; | |||
typedef enum aclrtFloatOverflowMode { | |||
ACL_RT_OVERFLOW_MODE_SATURATION = 0, | |||
ACL_RT_OVERFLOW_MODE_INFNAN, | |||
ACL_RT_OVERFLOW_MODE_UNDEF, | |||
} aclrtFloatOverflowMode; | |||
typedef struct tagRtGroupInfo aclrtGroupInfo; | |||
typedef struct rtExceptionInfo aclrtExceptionInfo; | |||
@@ -867,6 +873,20 @@ ACL_FUNC_VISIBILITY aclError aclrtSynchronizeStream(aclrtStream stream); | |||
/** | |||
* @ingroup AscendCL | |||
* @brief block the host until all tasks | |||
* in the specified stream have completed | |||
* | |||
* @param stream [IN] the stream to wait | |||
* @param timeout [IN] timeout value,the unit is milliseconds | |||
* -1 means waiting indefinitely, 0 means check whether synchronization is complete immediately | |||
* | |||
* @retval ACL_SUCCESS The function is successfully executed. | |||
* @retval OtherValues Failure | |||
*/ | |||
ACL_FUNC_VISIBILITY aclError aclrtSynchronizeStreamWithTimeout(aclrtStream stream, int32_t timeout); | |||
/** | |||
* @ingroup AscendCL | |||
* @brief Blocks the operation of the specified Stream until | |||
* the specified Event is completed. | |||
* Support for multiple streams waiting for the same event. | |||
@@ -1035,6 +1055,48 @@ ACL_FUNC_VISIBILITY aclError aclrtGetMemInfo(aclrtMemAttr attr, size_t *free, si | |||
*/ | |||
ACL_FUNC_VISIBILITY aclError aclrtSetOpWaitTimeout(uint32_t timeout); | |||
/** | |||
* @ingroup AscendCL | |||
* @brief enable or disable overflow switch on some stream | |||
* @param stream [IN] set overflow switch on this stream | |||
* @param flag [IN] 0 : disable 1 : enable | |||
* | |||
* @retval ACL_SUCCESS The function is successfully executed. | |||
* @retval OtherValues Failure | |||
*/ | |||
ACL_FUNC_VISIBILITY aclError aclrtSetStreamOverflowSwitch(aclrtStream stream, uint32_t flag); | |||
/** | |||
* @ingroup AscendCL | |||
* @brief get overflow switch on some stream | |||
* @param stream [IN] get overflow switch on this stream | |||
* @param flag [OUT] current overflow switch, 0 : disable others : enable | |||
* | |||
* @retval ACL_SUCCESS The function is successfully executed. | |||
* @retval OtherValues Failure | |||
*/ | |||
ACL_FUNC_VISIBILITY aclError aclrtGetStreamOverflowSwitch(aclrtStream stream, uint32_t *flag); | |||
/** | |||
* @ingroup AscendCL | |||
* @brief set saturation mode | |||
* @param mode [IN] target saturation mode | |||
* | |||
* @retval ACL_SUCCESS The function is successfully executed. | |||
* @retval OtherValues Failure | |||
*/ | |||
ACL_FUNC_VISIBILITY aclError aclrtSetDeviceSatMode(aclrtFloatOverflowMode mode); | |||
/** | |||
* @ingroup AscendCL | |||
* @brief get saturation mode | |||
* @param mode [OUT] get saturation mode | |||
* | |||
* @retval ACL_SUCCESS The function is successfully executed. | |||
* @retval OtherValues Failure | |||
*/ | |||
ACL_FUNC_VISIBILITY aclError aclrtGetDeviceSatMode(aclrtFloatOverflowMode *mode); | |||
#ifdef __cplusplus | |||
} | |||
#endif | |||
@@ -27,6 +27,10 @@ extern "C" { | |||
#define ACL_TDT_QUEUE_PERMISSION_DEQUEUE 2 | |||
#define ACL_TDT_QUEUE_PERMISSION_ENQUEUE 4 | |||
#define ACL_TDT_QUEUE_ROUTE_UNBIND 0 | |||
#define ACL_TDT_QUEUE_ROUTE_BIND 1 | |||
#define ACL_TDT_QUEUE_ROUTE_BIND_ABNORMAL 2 | |||
typedef void *acltdtBuf; | |||
typedef struct tagMemQueueAttr acltdtQueueAttr; | |||
typedef struct acltdtQueueRouteList acltdtQueueRouteList; | |||
@@ -43,8 +47,9 @@ typedef enum { | |||
typedef enum { | |||
ACL_TDT_QUEUE_ROUTE_QUERY_SRC = 0, | |||
ACL_TDT_QUEUE_ROUTE_QUERY_DST, | |||
ACL_TDT_QUEUE_ROUTE_QUERY_SRC_AND_DST | |||
ACL_TDT_QUEUE_ROUTE_QUERY_DST = 1, | |||
ACL_TDT_QUEUE_ROUTE_QUERY_SRC_AND_DST = 2, | |||
ACL_TDT_QUEUE_ROUTE_QUERY_ABNORMAL = 100 | |||
} acltdtQueueRouteQueryMode; | |||
typedef enum { | |||
@@ -261,6 +266,76 @@ ACL_FUNC_VISIBILITY aclError acltdtGetBufData(const acltdtBuf buf, void **dataPt | |||
/** | |||
* @ingroup AscendCL | |||
* @brief set data buf effective len | |||
* | |||
* @param buf [IN] acltdtBuf | |||
* @param len [IN] set effective len to data buf which must be smaller than size acquired by acltdtGetBufData | |||
* | |||
* @retval ACL_SUCCESS The function is successfully executed. | |||
* @retval OtherValues Failure | |||
* | |||
* @see acltdtGetBufData acltdtGetBufDataLen | |||
*/ | |||
ACL_FUNC_VISIBILITY aclError acltdtSetBufDataLen(acltdtBuf buf, size_t len); | |||
/** | |||
* @ingroup AscendCL | |||
* @brief get data buf effective len | |||
* | |||
* @param buf [IN] acltdtBuf | |||
* @param len [OUT] get effective len which is set by acltdtSetBufDataLen | |||
* | |||
* @retval ACL_SUCCESS The function is successfully executed. | |||
* @retval OtherValues Failure | |||
* | |||
* @see acltdtSetBufDataLen | |||
*/ | |||
ACL_FUNC_VISIBILITY aclError acltdtGetBufDataLen(acltdtBuf buf, size_t *len); | |||
/** | |||
* @ingroup AscendCL | |||
* @brief append acltdtBuf to acltdtBuf chain | |||
* | |||
* @param headBuf [IN] acltdtBuf chain head | |||
* @param buf [IN] acltdtBuf will be appended | |||
* | |||
* @retval ACL_SUCCESS The function is successfully executed. | |||
* @retval OtherValues Failure | |||
* | |||
*/ | |||
ACL_FUNC_VISIBILITY aclError acltdtAppendBufChain(acltdtBuf headBuf, acltdtBuf buf); | |||
/** | |||
* @ingroup AscendCL | |||
* @brief get acltdtBuf chain total size | |||
* | |||
* @param headBuf [IN] acltdtBuf chain head | |||
* @param num [OUT] acltdtBuf chain total size | |||
* | |||
* @retval ACL_SUCCESS The function is successfully executed. | |||
* @retval OtherValues Failure | |||
* | |||
* @see acltdtAppendBufChain | |||
*/ | |||
ACL_FUNC_VISIBILITY aclError acltdtGetBufChainNum(acltdtBuf headBuf, uint32_t *num); | |||
/** | |||
* @ingroup AscendCL | |||
* @brief get acltdtBuf from acltdtBuf chain by index | |||
* | |||
* @param headBuf [IN] acltdtBuf chain head | |||
* @param index [IN] the index which is smaller than num acquired from acltdtGetBufChainNum | |||
* @param buf [OUT] the acltdtBuf from acltdtBuf on index | |||
* | |||
* @retval ACL_SUCCESS The function is successfully executed. | |||
* @retval OtherValues Failure | |||
* | |||
* @see acltdtAppendBufChain acltdtGetBufChainNum | |||
*/ | |||
ACL_FUNC_VISIBILITY aclError acltdtGetBufFromChain(acltdtBuf headBuf, uint32_t index, acltdtBuf *buf); | |||
/** | |||
* @ingroup AscendCL | |||
* @brief get private data buf address and size | |||
* | |||
* @param buf [IN] acltdtBuf | |||
@@ -522,4 +597,4 @@ ACL_FUNC_VISIBILITY aclError acltdtSetQueueRouteQueryInfo(acltdtQueueRouteQueryI | |||
} | |||
#endif | |||
#endif // INC_EXTERNAL_ACL_ACL_TDT_QUEUE_H_ | |||
#endif // INC_EXTERNAL_ACL_ACL_TDT_QUEUE_H_ |
@@ -23,6 +23,7 @@ | |||
#include "ge/ge_api_error_codes.h" | |||
#include "ge/ge_api_types.h" | |||
#include "ge/ge_data_flow_api.h" | |||
#include "graph/graph.h" | |||
#include "graph/tensor.h" | |||
@@ -193,6 +194,36 @@ class GE_FUNC_VISIBILITY Session { | |||
uint64_t GetSessionId() const; | |||
/// @ingroup ge_graph | |||
/// @brief Feed input data to graph. | |||
/// @param [in] graph_id graph id | |||
/// @param [in] inputs input data | |||
/// @param [in] info intput data flow flag | |||
/// @param [in] timeout data feed timeout(ms), -1 means never timeout | |||
/// @return Status result of function | |||
Status FeedDataFlowGraph(uint32_t graph_id, const std::vector<Tensor> &inputs, const DataFlowInfo &info, | |||
int32_t timeout); | |||
/// @ingroup ge_graph | |||
/// @brief Fetch graph output data in order. | |||
/// @param [in] graph_id graph id | |||
/// @param [out] outputs output data | |||
/// @param [out] info output data flow flag | |||
/// @param [in] timeout data fetch timeout(ms), -1 means never timeout | |||
/// @return Status result of function | |||
Status FetchDataFlowGraph(uint32_t graph_id, std::vector<Tensor> &outputs, DataFlowInfo &info, int32_t timeout); | |||
/// @ingroup ge_graph | |||
/// @brief Fetch graph output data in order. | |||
/// @param [in] graph_id graph id | |||
/// @param [in] indexes fetch output data order(index cannot be duplicated) | |||
/// @param [out] outputs output data | |||
/// @param [out] info output data flow flag | |||
/// @param [in] timeout data ftech timeout(ms), -1 means never timeout | |||
/// @return Status result of function | |||
Status FetchDataFlowGraph(uint32_t graph_id, const std::vector<uint32_t> &indexes, std::vector<Tensor> &outputs, | |||
DataFlowInfo &info, int32_t timeout); | |||
private: | |||
uint64_t sessionId_; | |||
}; | |||
@@ -36,9 +36,9 @@ | |||
(static_cast<uint32_t>(0xFFU & (static_cast<uint32_t>(sysid))) << 17U) | \ | |||
(static_cast<uint32_t>(0xFFU & (static_cast<uint32_t>(modid))) << 12U) | \ | |||
(static_cast<uint32_t>(0x0FFFU) & (static_cast<uint32_t>(value))); \ | |||
const ErrorNoRegisterar g_errorno_##name((name), (desc)); | |||
const ErrorNoRegisterar g_errorno_##name((name), (desc)) | |||
#define GE_ERRORNO_EXTERNAL(name, desc) const ErrorNoRegisterar g_errorno_##name((name), (desc)); | |||
#define GE_ERRORNO_EXTERNAL(name, desc) const ErrorNoRegisterar g_errorno_##name((name), (desc)) | |||
namespace ge { | |||
class GE_FUNC_VISIBILITY StatusFactory { | |||
@@ -30,9 +30,6 @@ namespace ge { | |||
// Option key: graph run mode | |||
const char_t *const OPTION_GRAPH_RUN_MODE = "ge.graphRunMode"; | |||
const char_t *const OPTION_DEVICE_TYPE = "ge.deviceType"; | |||
// Option key: topo sorting mode | |||
const char *const OPTION_TOPO_SORTING_MODE = "ge.topoSortingMode"; | |||
// Option key: ome init | |||
const char_t *const OPTION_EXEC_SESSION_ID = "ge.exec.sessionId"; | |||
const char_t *const OPTION_EXEC_DEVICE_ID = "ge.exec.deviceId"; | |||
@@ -43,6 +40,9 @@ const char_t *const OPTION_EXEC_RANK_ID = "ge.exec.rankId"; | |||
const char_t *const OPTION_EXEC_POD_NAME = "ge.exec.podName"; | |||
const char_t *const OPTION_EXEC_DEPLOY_MODE = "ge.exec.deployMode"; | |||
const char_t *const OPTION_EXEC_RANK_TABLE_FILE = "ge.exec.rankTableFile"; | |||
const char_t *const OPTION_EXEC_RANK_TABLE = "ge.exec.rankTable"; | |||
const char_t *const OPTION_EXEC_HCOM_GROUPLIST = "ge.exec.hcomGrouplist"; | |||
const char_t *const OPTION_EXEC_HCOM_RANK_MAPPING = "ge.exec.hcomRankMapping"; | |||
const char_t *const GE_AICPU_FLAG = "ge.aicpuFlag"; | |||
const char_t *const OPTION_EXEC_EXTERN_PLUGIN_PATH = "ge.soLoadPath"; | |||
// Dump flag and para | |||
@@ -82,9 +82,12 @@ const char_t *const VARIABLE_MEMORY_MAX_SIZE = "ge.variableMemoryMaxSize"; | |||
const char_t *const OPTION_EXEC_REUSE_ZERO_COPY_MEMORY = "ge.exec.reuseZeroCopyMemory"; | |||
const std::string ATOMIC_CLEAN_POLICY = "ge.exec.atomicCleanPolicy"; | |||
const std::string ATOMIC_CLEAN_POLICY_SUB = "atomic_clean_policy"; | |||
const char_t *const OPTION_EXEC_LOGICAL_DEVICE_CLUSTER_DEPLOY_MODE = "ge.exec.logicalDeviceClusterDeployMode"; | |||
const char_t *const OPTION_EXEC_LOGICAL_DEVICE_ID = "ge.exec.logicalDeviceId"; | |||
const char_t *const OPTION_EXEC_MODEL_DEPLOY_MODE = "ge.exec.modelDeployMode"; | |||
const char_t *const OPTION_EXEC_MODEL_DEPLOY_DEVICELIST = "ge.exec.modelDeployDevicelist"; | |||
namespace configure_option { | |||
const char_t *const STREAM_NUM = "ge.streamNum"; | |||
@@ -134,6 +137,8 @@ const char_t *const OP_PRECISION_MODE = "ge.exec.op_precision_mode"; | |||
const char_t *const CUSTOMIZE_DTYPES = "ge.customizeDtypes"; | |||
const char_t *const COMPRESSION_OPTIMIZE_CONF = "ge.compressionOptimizeConf"; | |||
const char_t *const OP_DEBUG_CONFIG = "op_debug_config"; | |||
const char_t *const ATOMIC_CLEAN_POLICY = "ge.exec.atomicCleanPolicy"; | |||
const char_t *const ATOMIC_CLEAN_POLICY_SUB = "atomic_clean_policy"; | |||
} // namespace configure_option | |||
// Configure stream num by Session constructor options param, | |||
// its value should be int32_t type, default value is "1" | |||
@@ -389,18 +394,8 @@ const std::string OP_EXECUTE_TIMEOUT = "ge.exec.opExecuteTimeout"; | |||
const char_t *const FILE_CONSTANT_PATH = "ge.exec.value_bins"; | |||
const char_t *const ENABLE_GRAPH_PARALLEL = "ge.enableGraphParallel"; | |||
const char_t *const RESOURCE_CONFIG_PATH = "ge.resourceConfigPath"; | |||
const std::string RECOMPUTE = "ge.recompute"; | |||
const char_t *const GRAPH_PARALLEL_OPTION_PATH = "ge.graphParallelOptionPath"; | |||
// Graph run mode | |||
enum GraphRunMode { PREDICTION = 0, TRAIN }; | |||
// Topo sorting mode | |||
enum class TopoSortingMode { BFS = 0, DFS = 1 }; | |||
// Input/Output tensor info | |||
struct InputTensorInfo { | |||
uint32_t data_type; // data type | |||
@@ -486,6 +481,8 @@ static const char_t *const CUSTOMIZE_DTYPES = "ge.customizeDtypes"; | |||
static const char_t *const COMPRESSION_OPTIMIZE_CONF = "ge.compressionOptimizeConf"; | |||
static const char_t *const INPUT_DATA_NAMES = "input_data_names"; | |||
static const char_t *const OP_DEBUG_CONFIG = "op_debug_config"; | |||
static const char_t *const ATOMIC_CLEAN_POLICY = "ge.exec.atomicCleanPolicy"; | |||
static const char_t *const ATOMIC_CLEAN_POLICY_SUB = "atomic_clean_policy"; | |||
// for interface: aclgrphBuildModel | |||
#ifdef __GNUC__ | |||
@@ -518,7 +515,8 @@ const std::set<std::string> ir_builder_suppported_options = {INPUT_FORMAT, | |||
MODIFY_MIXLIST, | |||
CUSTOMIZE_DTYPES, | |||
BUILD_INNER_MODEL, | |||
OP_DEBUG_CONFIG}; | |||
OP_DEBUG_CONFIG, | |||
EXCLUDE_ENGINES}; | |||
// for interface: aclgrphParse | |||
const std::set<std::string> ir_parser_suppported_options = { | |||
@@ -0,0 +1,60 @@ | |||
/** | |||
* Copyright 2019-2022 Huawei Technologies Co., Ltd | |||
* | |||
* Licensed under the Apache License, Version 2.0 (the "License"); | |||
* you may not use this file except in compliance with the License. | |||
* You may obtain a copy of the License at | |||
* | |||
* http://www.apache.org/licenses/LICENSE-2.0 | |||
* | |||
* Unless required by applicable law or agreed to in writing, software | |||
* distributed under the License is distributed on an "AS IS" BASIS, | |||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||
* See the License for the specific language governing permissions and | |||
* limitations under the License. | |||
*/ | |||
#ifndef INC_EXTERNAL_GE_GE_DATA_FLOW_API_H | |||
#define INC_EXTERNAL_GE_GE_DATA_FLOW_API_H | |||
#include <memory> | |||
#include "ge_error_codes.h" | |||
namespace ge { | |||
enum class DataFlowFlag : uint32_t { | |||
DATA_FLOW_FLAG_EOS = (1U << 0U), // data flag end | |||
DATA_FLOW_FLAG_SEG = (1U << 1U) // segment flag for discontinuous | |||
}; | |||
class DataFlowInfoImpl; | |||
class GE_FUNC_VISIBILITY DataFlowInfo { | |||
public: | |||
DataFlowInfo(); | |||
~DataFlowInfo(); | |||
DataFlowInfo(const DataFlowInfo &context) = delete; | |||
DataFlowInfo(const DataFlowInfo &&context) = delete; | |||
DataFlowInfo &operator=(const DataFlowInfo &context) = delete; | |||
DataFlowInfo &operator=(const DataFlowInfo &&context) = delete; | |||
void SetStartTime(const uint64_t start_time); | |||
uint64_t GetStartTime() const; | |||
void SetEndTime(const uint64_t end_time); | |||
uint64_t GetEndTime() const; | |||
/** | |||
* @brief Set the Flow Flags object. | |||
* @param flow_flags can use operate | to merge multi DataFlowFla to flags. | |||
*/ | |||
void SetFlowFlags(const uint32_t flow_flags); | |||
/** | |||
* @brief Get the Flow Flags object. | |||
* @return uint32_t flow flags, can use operate & with DataFlowFlag to check which bit is set. | |||
*/ | |||
uint32_t GetFlowFlags() const; | |||
private: | |||
std::shared_ptr<DataFlowInfoImpl> impl_; | |||
}; | |||
} // namespace ge | |||
#endif // INC_EXTERNAL_GE_GE_DATA_FLOW_API_H |
@@ -14,11 +14,6 @@ | |||
* limitations under the License. | |||
*/ | |||
/** | |||
* @file hccl.h | |||
* @brief HCCL API | |||
*/ | |||
#ifndef HCCL_H_ | |||
#define HCCL_H_ | |||
@@ -66,8 +61,8 @@ extern HcclResult HcclCommInitRootInfo(uint32_t nRanks, const HcclRootInfo *root | |||
* @param sendBuf A pointer identifying the input data address of the operator. | |||
* @param recvBuf A pointer identifying the output data address of the operator. | |||
* @param count An integer(u64) identifying the number of the output data. | |||
* @param dataType The data type of the operator, must be one of the following types: int8, int16, int32, float16, | |||
* float32. | |||
* @param dataType The data type of the operator, must be one of the following types: int8, int16, int32, | |||
float16, float32, bfloat16. | |||
* @param op The reduction type of the operator, must be one of the following types: sum, min, max, prod. | |||
* @param comm A pointer identifying the communication resource based on. | |||
* @param stream A pointer identifying the stream information. | |||
@@ -82,7 +77,7 @@ extern HcclResult HcclAllReduce(void *sendBuf, void *recvBuf, uint64_t count, Hc | |||
* @param buf A pointer identifying the data address of the operator. | |||
* @param count An integer(u64) identifying the number of the data. | |||
* @param dataType The data type of the operator, must be one of the following types: int8, int16, int32, int64, | |||
uint8, uint16, uint32, uint64, float16, float32, float64. | |||
uint8, uint16, uint32, uint64, float16, float32, float64, bfloat16. | |||
* @param root An integer(u32) identifying the the root rank in the operator. | |||
* @param comm A pointer identifying the communication resource based on | |||
* @param stream A pointer identifying the stream information. | |||
@@ -97,7 +92,8 @@ extern HcclResult HcclBroadcast(void *buf, uint64_t count, HcclDataType dataType | |||
* @param sendBuf A pointer identifying the input data address of the operator. | |||
* @param recvBuf A pointer identifying the output data address of the operator. | |||
* @param recvCount An integer(u64) identifying the number of the output data. | |||
* @param dataType The data type of the operator, must be one of the following types: int8, int32, float16, float32. | |||
* @param dataType The data type of the operator, must be one of the following types: int8, int32, | |||
float16, float32, bfloat16. | |||
* @param op The reduction type of the operator, must be one of the following types: sum, min, max, prod. | |||
* @param comm A pointer identifying the communication resource based on. | |||
* @param stream A pointer identifying the stream information. | |||
@@ -113,7 +109,7 @@ extern HcclResult HcclReduceScatter(void *sendBuf, void *recvBuf, uint64_t recvC | |||
* @param recvBuf A pointer identifying the output data address of the operator. | |||
* @param sendCount An integer(u64) identifying the number of the input data. | |||
* @param dataType The data type of the operator, must be one of the following types: int8, int16, int32, int64, | |||
uint8, uint16, uint32, uint64, float16, float32, float64. | |||
uint8, uint16, uint32, uint64, float16, float32, float64, bfloat16. | |||
* @param comm A pointer identifying the communication resource based on. | |||
* @param stream A pointer identifying the stream information. | |||
* @return HcclResult | |||
@@ -152,7 +148,7 @@ extern HcclResult HcclBarrier(HcclComm comm, aclrtStream stream); | |||
* @param sendBuff A pointer identifying the input data address of the operator. | |||
* @param count An integer(u64) identifying the number of the send data. | |||
* @param dataType The data type of the operator, must be one of the following types: int8, int16, int32, int64, | |||
uint8, uint16, uint32, uint64, float16, float32, float64. | |||
uint8, uint16, uint32, uint64, float16, float32, float64, bfloat16. | |||
* @param destRank An integer identifying the destination rank. | |||
* @param comm A pointer identifying the communication resource based on. | |||
* @param stream A pointer identifying the stream information. | |||
@@ -166,7 +162,7 @@ extern HcclResult HcclSend(void *sendBuf, uint64_t count, HcclDataType dataType, | |||
* @param recvBuff A pointer identifying the output data address of the operator. | |||
* @param count An integer(u64) identifying the number of the receive data. | |||
* @param dataType The data type of the operator, must be one of the following types: int8, int16, int32, int64, | |||
uint8, uint16, uint32, uint64, float16, float32, float64. | |||
uint8, uint16, uint32, uint64, float16, float32, float64, bfloat16. | |||
* @param srcRank An integer identifying the source rank. | |||
* @param comm A pointer identifying the communication resource based on. | |||
* @param stream A pointer identifying the stream information. | |||
@@ -183,7 +179,7 @@ extern HcclResult HcclRecv(void *recvBuf, uint64_t count, HcclDataType dataType, | |||
* @param sdispls Integer array, where entry i specifies the displacement (offset from sendbuf, in units of sendtype) | |||
from which to send data to rank i. | |||
* @param sendType Datatype of send buffer elements, must be one of the following types: int8, int16, int32, int64, | |||
uint8, uint16, uint32, uint64, float16, float32, float64. | |||
uint8, uint16, uint32, uint64, float16, float32, float64, bfloat16. | |||
* @param recvBuf A pointer identifying the output data address of the operator. | |||
* @param recvCounts Integer array, where entry j specifies the number of elements to receive from rank j. | |||
* @param rdispls Integer array, where entry j specifies the displacement (offset from recvbuf, in units of recvtype) to | |||
@@ -206,7 +202,7 @@ extern HcclResult HcclAlltoAllV(const void *sendBuf, const void *sendCounts, con | |||
* @param recvBuf A pointer identifying the output data address of the operator. | |||
* @param count An integer(u64) identifying the number of the output data. | |||
* @param dataType The data type of the operator, must be one of the following types: int8, int16, int32, float16, | |||
* float32. | |||
* float32, bfloat16. | |||
* @param op The reduction type of the operator, must be one of the following types: sum, min, max, prod. | |||
* @param root An integer(u32) identifying the the root rank in the operator. | |||
* @param comm A pointer identifying the communication resource based on. | |||
@@ -14,12 +14,6 @@ | |||
* limitations under the License. | |||
*/ | |||
/** | |||
* @file hccl_types.h | |||
* @brief HCCL data type definition | |||
* | |||
*/ | |||
#ifndef HCCL_TYPES_H_ | |||
#define HCCL_TYPES_H_ | |||
@@ -35,6 +35,7 @@ extern "C" { | |||
#endif | |||
#define GE_MODULE_NAME static_cast<int32_t>(GE) | |||
#define GE_MODULE_NAME_U16 static_cast<uint16_t>(GE) | |||
// trace status of log | |||
enum TraceStatus { TRACE_INIT = 0, TRACE_RUNNING, TRACE_WAITING, TRACE_STOP }; | |||
@@ -89,6 +90,9 @@ inline bool IsLogEnable(const int32_t module_name, const int32_t log_level) { | |||
do { \ | |||
dlog_event((RUN_LOG_MASK | GE_MODULE_NAME), "%" PRIu64 " %s:" fmt, GeLog::GetTid(), &__FUNCTION__[0], \ | |||
##__VA_ARGS__); \ | |||
if (IsLogEnable(GE_MODULE_NAME, DLOG_INFO)) { \ | |||
dlog_info(GE_MODULE_NAME, "%" PRIu64 " %s:" fmt, GeLog::GetTid(), &__FUNCTION__[0], ##__VA_ARGS__); \ | |||
} \ | |||
} while (false) | |||
#define GELOGT(VALUE, fmt, ...) \ | |||
@@ -110,6 +110,16 @@ | |||
} \ | |||
} while (false) | |||
// If expr is true, print info log and return the specified status | |||
#define GE_CHK_BOOL_RET_SPECIAL_STATUS(expr, _status, ...) \ | |||
do { \ | |||
const bool b = (expr); \ | |||
if (b) { \ | |||
GELOGI(__VA_ARGS__); \ | |||
return (_status); \ | |||
} \ | |||
} while (false) | |||
// If expr is not true, print the log and return the specified status | |||
#define GE_CHK_BOOL_RET_STATUS_NOLOG(expr, _status, ...) \ | |||
do { \ | |||
@@ -141,13 +151,13 @@ | |||
// If expr is not RT_ERROR_NONE, print the log and execute the exec_expr expression | |||
#define GE_CHK_RT_EXEC(expr, exec_expr) \ | |||
do { \ | |||
{ \ | |||
const rtError_t _rt_ret = (expr); \ | |||
if (_rt_ret != RT_ERROR_NONE) { \ | |||
GELOGE(ge::RT_FAILED, "Call rt api failed, ret: 0x%X", _rt_ret); \ | |||
exec_expr; \ | |||
} \ | |||
} while (false) | |||
} | |||
// If expr is not RT_ERROR_NONE, print the log and return | |||
#define GE_CHK_RT_RET(expr) \ | |||
@@ -67,7 +67,10 @@ const char_t *const GE_OPTION_EXEC_PLACEMENT = "ge.exec.placement"; | |||
// profiling data | |||
const std::string kTaskTypeAicore = "AI_CORE"; | |||
const std::string kTaskTypeMixAic = "MIX_AIC"; | |||
const std::string kTaskTypeMixAiv = "MIX_AIV"; | |||
const std::string kTaskTypeAicpu = "AI_CPU"; | |||
const std::string kTaskTypeDsa = "DSA"; | |||
const std::string kTaskTypeWriteBackData = "WRITE_BACK"; | |||
const std::string kTaskTypeInvalidData = "INVALID"; | |||
const std::string kTaskTypeInvalid = "TASK_TYPE_INVALID"; | |||
@@ -81,8 +84,11 @@ const std::string kEngineNameGeLocal = "DNN_VM_GE_LOCAL_OP_STORE"; | |||
const std::string kEngineNameAiCpu = "aicpu_ascend_kernel"; | |||
const std::string kEngineNameAiCpuTf = "aicpu_tf_kernel"; | |||
const std::string kEngineNameAiCore = "AIcoreEngine"; | |||
const std::string kEngineNameDvpp = "dvpp_ops_kernel"; | |||
const std::string kEngineNameDsa = "DSAEngine"; | |||
const std::string kAtomicOpType = "DynamicAtomicAddrClean"; | |||
const char_t *const kAICpuKernelLibName = "aicpu_kernel_lib_name"; | |||
const char *const kPartiallySupported = "partially_supported"; | |||
const std::string kShapeTypeStatic = "static"; | |||
const std::string kShapeTypeDynamic = "dynamic"; | |||
const std::string kAtomicPrefix = "_atomic"; | |||
@@ -387,5 +393,23 @@ struct ModelQueueParam { | |||
// internal options | |||
// 1: Graph resource evaluation does not limit model memory size. | |||
const char_t *const EVALUATE_GRAPH_RESOURCE_MODE = "ge.evaluateGraphResourceMode"; | |||
// 2: Enable graph parallel options | |||
const char_t *const ENABLE_GRAPH_PARALLEL = "ge.enableGraphParallel"; | |||
// 3: Config all resource and device mesh | |||
const char_t *const RESOURCE_CONFIG_PATH = "ge.resourceConfigPath"; | |||
// 4: Config graph parallel options path(should specific file name) | |||
const char_t *const GRAPH_PARALLEL_OPTION_PATH = "ge.graphParallelOptionPath"; | |||
// 5: auto recompute attribute | |||
const char_t *const RECOMPUTE = "ge.recompute"; | |||
// 6: Topological Sorting Mode | |||
const char_t *const OPTION_TOPOSORTING_MODE = "ge.topoSortingMode"; | |||
const std::set<std::string> ir_builder_suppported_options_inner = { | |||
EVALUATE_GRAPH_RESOURCE_MODE, ENABLE_GRAPH_PARALLEL, RESOURCE_CONFIG_PATH, GRAPH_PARALLEL_OPTION_PATH, RECOMPUTE, | |||
OPTION_TOPOSORTING_MODE}; | |||
} // namespace ge | |||
#endif // INC_FRAMEWORK_COMMON_GE_TYPES_H_ |
@@ -46,6 +46,7 @@ enum class ccKernelType { | |||
AI_CPU = 6, /* aicpu */ | |||
CUST_AI_CPU = 7, /* custom aicpu */ | |||
HOST_CPU = 8, /* host cpu */ | |||
DVPP = 9, /* dvpp */ | |||
INVALID = 10000 /* unknown kernel type */ | |||
}; | |||
@@ -222,12 +222,10 @@ namespace ge { | |||
*/ | |||
constexpr int32_t OM_PROTO_VERSION = 2; | |||
/// | |||
/// @ingroup domi_common | |||
/// @brief onverts Vector of a number to a string. | |||
/// @param [in] v Vector of a number | |||
/// @return string | |||
/// | |||
template <typename T> | |||
GE_FUNC_VISIBILITY std::string ToString(const std::vector<T> &v) { | |||
bool first = true; | |||
@@ -277,12 +275,9 @@ GE_FUNC_VISIBILITY std::string CurrentTimeInStr(); | |||
/// @return Timestamp, in microseconds (US) | |||
GE_FUNC_VISIBILITY uint64_t GetCurrentTimestamp(); | |||
/// | |||
/// @ingroup domi_common | |||
/// @brief Obtains the absolute time (timestamp) of the current system. | |||
/// @return Timestamp, in seconds (US) | |||
/// | |||
/// | |||
GE_FUNC_VISIBILITY uint32_t GetCurrentSecondTimestap(); | |||
/// @ingroup domi_common | |||
@@ -236,7 +236,7 @@ class GE_FUNC_VISIBILITY GeExecutor { | |||
/// @return: 0 for success / others for fail | |||
/// | |||
Status LoadModelWithQ(uint32_t &model_id, const std::shared_ptr<GeRootModel> &root_model, | |||
const ModelQueueParam &model_queue_param); | |||
const ModelQueueParam &model_queue_param) const; | |||
/// | |||
/// @ingroup ge | |||
@@ -109,8 +109,7 @@ class GE_FUNC_VISIBILITY GeGenerator { | |||
Graph &graph, std::vector<std::pair<std::string, std::string>> &inputs_name_type) const; | |||
Status BuildOriginalGraphInfo(OpDescPtr &op_desc, const std::vector<GeTensor> &inputs, | |||
const std::vector<GeTensor> &outputs, const std::string &model_file_name, | |||
bool is_offline, int32_t compile_flag, GraphStage graph_stage, Graph &graph, | |||
ComputeGraphPtr &compute_graph, bool &fuzz_compile_flag, | |||
bool is_offline, GraphStage graph_stage, Graph &graph, ComputeGraphPtr &compute_graph, | |||
std::vector<std::pair<std::string, std::string>> &inputs_name_type); | |||
private: | |||
@@ -125,7 +124,7 @@ class GE_FUNC_VISIBILITY GeGenerator { | |||
Status CheckForSingleOp(const OpDescPtr &op_desc, const std::vector<GeTensor> &inputs, | |||
const std::vector<GeTensor> &outputs) const; | |||
Status InferFormatForSingleOp(const OpDescPtr &op_desc, const Graph &graph) const; | |||
Status ResetAiCpuToDynamicShape(const ComputeGraphPtr &graph); | |||
Status ResetAiCpuToDynamicShape(const ComputeGraphPtr &graph) const; | |||
using GeRootModelPtr = std::shared_ptr<ge::GeRootModel>; | |||
Status SetModelNameForDump(const GeRootModelPtr &ge_root_model); | |||
@@ -133,9 +132,13 @@ class GE_FUNC_VISIBILITY GeGenerator { | |||
const std::vector<GeTensor> &outputs, | |||
const std::vector<std::pair<std::string, std::string>> &inputs_name_type, | |||
std::vector<ge::NamedAttrs> &generalized_build_attrs) const; | |||
void AddExcludeEnginesOption(const OpDescPtr &op_desc, std::map<std::string, std::string> &graph_options) const; | |||
void AddShapeGeneralizedOption(std::map<std::string, std::string> &graph_options); | |||
void SetFuzzCompile(const std::vector<GeTensor> &inputs, int32_t compile_flag); | |||
bool IsFuzzCompileEnable(); | |||
void ConvertOpInfosToOptions(const OpDescPtr &op_desc); | |||
class Impl; | |||
std::shared_ptr<Impl> impl_; | |||
}; | |||
} // namespace ge | |||
@@ -14,8 +14,8 @@ | |||
* limitations under the License. | |||
*/ | |||
#ifndef PARSER_COMMON_TYPES_H_ | |||
#define PARSER_COMMON_TYPES_H_ | |||
#ifndef INC_FRAMEWORK_OMG_PARSER_TYPES_H_ | |||
#define INC_FRAMEWORK_OMG_PARSER_TYPES_H_ | |||
#include <cstdint> | |||
#include <string> | |||
@@ -41,428 +41,428 @@ | |||
namespace ge { | |||
namespace parser { | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *DATA; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *AIPPDATA; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *CONVOLUTION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *CORRELATION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *CORRELATIONV2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *DECONVOLUTION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *POOLING; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ELTWISE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *RELU; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *RELU6; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SIGMOID; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ABSVAL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *TANH; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *PRELU; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *BATCHNORM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *FUSIONBATCHNORM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SCALE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *FULL_CONNECTION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SOFTMAX; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *PLUS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ACTIVATION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *FLATTEN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ADD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SUB; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *MUL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *MATMUL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *RSQRT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *BIASADD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *RESHAPE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *REFORMAT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *DEPCONVOLUTION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *DROPOUT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *DROPOUTGENMASK; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *DROPOUTDOMASK; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *CONCAT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ROIPOOLING; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *PROPOSAL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *FSRDETECTIONOUTPUT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *DETECTIONPOSTPROCESS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *LRN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *TRANSDATA; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *PERMUTE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SSDNORMALIZE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SSDPRIORBOX; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *NETOUTPUT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SSDDETECTIONOUTPUT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *REFINEDETDETECTIONOUTPUT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *CHANNELAXPY; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *PSROIPOOLING; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *POWER; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *POW; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ROIALIGN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *PYTHON; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *FREESPACEEXTRACT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SPATIALTF; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SHAPE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SHAPEN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ARGMAX; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *GATHERND; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *GATHER; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *REALDIV; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *PACK; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SLICE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SLICED; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *FLOORDIV; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SQUEEZE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *UNSQUEEZE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *STRIDEDSLICE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *RANGE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *RPNPROPOSALS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *DECODEBBOX; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *PAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *PADV2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *MIRRORPAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *TILE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SIZE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *CLIPBOXES; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *FASTRCNNPREDICTIONS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SPLIT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SPLITV; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *EXPANDDIMS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *EMPTY; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *MEAN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *GREATER; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SWITCH; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SWITCHN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *MERGE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SYMBOLICGRADIENT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *REMOTECALL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *_IF; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *STATELESSIF; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *IF; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *CASE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *STATELESSCASE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *_WHILE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *WHILE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *STATELESSWHILE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *FOR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *PARTITIONEDCALL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *STATEFULPARTITIONEDCALL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *FAKEPARAM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *TRANSPOSE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *TRANSPOSED; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *CAST; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *REGION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *YOLO; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *YOLODETECTIONOUTPUT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *FILL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *REVERSE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *UNPACK; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *YOLO2REORG; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *REDUCESUM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SUM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *CONSTANT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *FILECONSTANT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *RESIZEBILINEAR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *RESIZEBILINEARGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *MAXIMUM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *FRAMEWORKOP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ARG; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *FUSEDBATCHNORMGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *LSTM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *HIGHWAY; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *RNN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ATTENTIONDECODER; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *LOGICAL_NOT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *LOGICAL_AND; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *LOGICAL_OR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *EQUAL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *NOTEQUAL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *INTERP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SHUFFLECHANNEL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *AIPP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *MULTISHAPE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *RECIPROCAL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SELU; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ELU; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ACOSH; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ASINH; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *MINIMUM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *CLIP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *L2NORMALIZE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *CROPANDRESIZE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *UNUSEDCONST; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SPARSETODENSE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *NONMAXSUPPRESSION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *TOPKV2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *INVERTPERMUTATION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *MULTINOMIAL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *REVERSESEQUENCE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *REDUCEPROD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *REDUCEMAX; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *REDUCEMIN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *EXTRACTIMAGEPATCHES; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SQRT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *REDUCEALL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *RESIZENEARESTNEIGHBOR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SPACETOBATCHND; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *BATCHTOSPACEND; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ASSERT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *GREATEREQUAL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *FLOOR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *RANDOMUNIFORM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *BATCHMATMUL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SPACETODEPTH; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *DEPTHTOSPACE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *RINT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ATAN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ATAN2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ATANH; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ACOS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ASIN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *NEG; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *LOG; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *TAN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ROUND; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *UPSAMPLE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *FLOORMOD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *LESS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *LESSEQUAL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ONEHOT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *REFSWITCH; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *REFMERGE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ENTER; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *REFENTER; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *LOOPCOND; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *NEXTITERATION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *REFNEXTITERATION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *EXIT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *REFEXIT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *CONTROLTRIGGER; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ZEROSLIKE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *EXP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *WHERE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *FAKEQUANTWITHMINMAXVARS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SOFTPLUS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SOFTSIGN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *COSH; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SINH; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SQUAREDDIFFERENCE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const DATA; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const AIPPDATA; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const CONVOLUTION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const CORRELATION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const CORRELATIONV2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const DECONVOLUTION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const POOLING; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ELTWISE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const RELU; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const RELU6; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SIGMOID; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ABSVAL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const TANH; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const PRELU; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const BATCHNORM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const FUSIONBATCHNORM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SCALE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const FULL_CONNECTION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SOFTMAX; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const PLUS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ACTIVATION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const FLATTEN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ADD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SUB; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const MUL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const MATMUL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const RSQRT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const BIASADD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const RESHAPE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const REFORMAT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const DEPCONVOLUTION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const DROPOUT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const DROPOUTGENMASK; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const DROPOUTDOMASK; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const CONCAT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ROIPOOLING; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const PROPOSAL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const FSRDETECTIONOUTPUT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const DETECTIONPOSTPROCESS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const LRN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const TRANSDATA; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const PERMUTE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SSDNORMALIZE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SSDPRIORBOX; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const NETOUTPUT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SSDDETECTIONOUTPUT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const REFINEDETDETECTIONOUTPUT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const CHANNELAXPY; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const PSROIPOOLING; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const POWER; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const POW; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ROIALIGN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const PYTHON; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const FREESPACEEXTRACT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SPATIALTF; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SHAPE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SHAPEN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ARGMAX; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const GATHERND; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const GATHER; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const REALDIV; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const PACK; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SLICE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SLICED; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const FLOORDIV; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SQUEEZE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const UNSQUEEZE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const STRIDEDSLICE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const RANGE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const RPNPROPOSALS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const DECODEBBOX; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const PAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const PADV2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const MIRRORPAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const TILE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SIZE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const CLIPBOXES; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const FASTRCNNPREDICTIONS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SPLIT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SPLITV; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const EXPANDDIMS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const EMPTY; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const MEAN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const GREATER; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SWITCH; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SWITCHN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const MERGE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SYMBOLICGRADIENT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const REMOTECALL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const _IF; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const STATELESSIF; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const IF; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const CASE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const STATELESSCASE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const _WHILE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const WHILE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const STATELESSWHILE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const FOR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const PARTITIONEDCALL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const STATEFULPARTITIONEDCALL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const FAKEPARAM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const TRANSPOSE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const TRANSPOSED; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const CAST; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const REGION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const YOLO; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const YOLODETECTIONOUTPUT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const FILL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const REVERSE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const UNPACK; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const YOLO2REORG; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const REDUCESUM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SUM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const CONSTANT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const FILECONSTANT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const RESIZEBILINEAR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const RESIZEBILINEARGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const MAXIMUM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const FRAMEWORKOP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ARG; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const FUSEDBATCHNORMGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const LSTM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const HIGHWAY; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const RNN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ATTENTIONDECODER; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const LOGICAL_NOT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const LOGICAL_AND; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const LOGICAL_OR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const EQUAL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const NOTEQUAL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const INTERP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SHUFFLECHANNEL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const AIPP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const MULTISHAPE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const RECIPROCAL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SELU; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ELU; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ACOSH; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ASINH; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const MINIMUM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const CLIP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const L2NORMALIZE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const CROPANDRESIZE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const UNUSEDCONST; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SPARSETODENSE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const NONMAXSUPPRESSION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const TOPKV2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const INVERTPERMUTATION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const MULTINOMIAL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const REVERSESEQUENCE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const REDUCEPROD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const REDUCEMAX; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const REDUCEMIN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const EXTRACTIMAGEPATCHES; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SQRT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const REDUCEALL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const RESIZENEARESTNEIGHBOR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SPACETOBATCHND; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const BATCHTOSPACEND; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ASSERT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const GREATEREQUAL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const FLOOR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const RANDOMUNIFORM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const BATCHMATMUL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SPACETODEPTH; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const DEPTHTOSPACE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const RINT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ATAN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ATAN2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ATANH; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ACOS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ASIN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const NEG; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const LOG; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const TAN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ROUND; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const UPSAMPLE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const FLOORMOD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const LESS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const LESSEQUAL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ONEHOT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const REFSWITCH; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const REFMERGE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ENTER; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const REFENTER; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const LOOPCOND; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const NEXTITERATION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const REFNEXTITERATION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const EXIT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const REFEXIT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const CONTROLTRIGGER; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ZEROSLIKE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const EXP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const WHERE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const FAKEQUANTWITHMINMAXVARS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SOFTPLUS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SOFTSIGN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const COSH; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SINH; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SQUAREDDIFFERENCE; | |||
// for retinanet scope fusion | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *REQUIREDSPACETOBATCHPADDINGS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SSDPOSTPROCESSOR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *RETINANETBOXES; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *RETINAMULTIANCHORS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *RETINANETCLIPPEDBOXES; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *RETINANETFILTEREDDETECTIONS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *RETINANETPOSTPROCESSOR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *RETINANETANCHORS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *FASTERRCNNMAP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *FASTERRCNNMAP1; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *FASTERRCNNSECONDSTAGEPOSTPROCESSOR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *FASTERRCNNROIINTERPOOLING; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *FASTERRCNNFIRSTSTAGEPOSTPROCESSOR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *FASTERRCNNGRIDANCHORGENERATOR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ROIINTERPOOLING; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *FASTERRCNNCLIPTOWINDOW; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *EMBEDLOOKUP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *HASHLOOKUP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *LSH_PROJ; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SVDF; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SSDANCHORGENERATOR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *IDENTITY; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *IDENTITYN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *PLACEHOLDERWITHDEFAULT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SELECT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *GETSPAN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *STOPGRADIENT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *PREVENTGRADIENT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *GUARANTEECONST; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *BROADCASTGRADIENTARGS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *BROADCASTARGS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *CONFUSIONMATRIX; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *RANK; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *PLACEHOLDER; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *END; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *BASICLSTMCELL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *GETNEXT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *INITDATA; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *REFIDENTITY; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *BITCAST; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const REQUIREDSPACETOBATCHPADDINGS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SSDPOSTPROCESSOR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const RETINANETBOXES; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const RETINAMULTIANCHORS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const RETINANETCLIPPEDBOXES; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const RETINANETFILTEREDDETECTIONS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const RETINANETPOSTPROCESSOR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const RETINANETANCHORS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const FASTERRCNNMAP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const FASTERRCNNMAP1; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const FASTERRCNNSECONDSTAGEPOSTPROCESSOR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const FASTERRCNNROIINTERPOOLING; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const FASTERRCNNFIRSTSTAGEPOSTPROCESSOR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const FASTERRCNNGRIDANCHORGENERATOR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ROIINTERPOOLING; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const FASTERRCNNCLIPTOWINDOW; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const EMBEDLOOKUP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const HASHLOOKUP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const LSH_PROJ; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SVDF; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SSDANCHORGENERATOR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const IDENTITY; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const IDENTITYN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const PLACEHOLDERWITHDEFAULT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SELECT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const GETSPAN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const STOPGRADIENT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const PREVENTGRADIENT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const GUARANTEECONST; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const BROADCASTGRADIENTARGS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const BROADCASTARGS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const CONFUSIONMATRIX; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const RANK; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const PLACEHOLDER; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const END; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const BASICLSTMCELL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const GETNEXT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const INITDATA; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const REFIDENTITY; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const BITCAST; | |||
/***************Ann special operator*************************/ | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ANN_MEAN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ANN_CONVOLUTION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ANN_DEPCONVOLUTION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ANN_FULLCONNECTION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ANN_NETOUTPUT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ANN_DATA; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ANN_RESHAPE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ANN_ADD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ANN_MUL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ANN_SUB; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ANN_DIV; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ANN_DEQUANTIZE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ANN_QUANTIZE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ANN_PAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ANN_RESIZE_BILINEAR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ANN_MEAN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ANN_CONVOLUTION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ANN_DEPCONVOLUTION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ANN_FULLCONNECTION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ANN_NETOUTPUT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ANN_DATA; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ANN_RESHAPE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ANN_ADD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ANN_MUL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ANN_SUB; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ANN_DIV; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ANN_DEQUANTIZE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ANN_QUANTIZE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ANN_PAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ANN_RESIZE_BILINEAR; | |||
/***************************************************/ | |||
/******************Training operator*************************/ | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *GATHERV2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *CONVGRADFILTER; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *CONV2D; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *CONV2DBACKPROPINPUT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *FUSEDBATCHNORM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *BIASADDGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ACTIVATIONGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *MAXPOOLWITHARGMAX; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *MAXPOOLGRADWITHARGMAX; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SPARSESOFTMAXCROSSENTROPYWITHLOGITS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SNAPSHOT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *VAR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *MEANGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *TRANSLATE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ADDN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *L2LOSS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *MULTIPLY; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *HUBERLOSSGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *HUBERLOSS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *NEGATIVE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SSDCAST; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SPARSESOFTMAXCROSSENTROPY; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SPARSESOFTMAXCROSSENTROPYGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SSDSQUEEZEFUSION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *CONCATFOUR2FIVE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *CONCATFIVE2FOUR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SSDREALDIVTILEMUL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SSDSUMMULREALDIVMEAN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *VARIABLEV2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *VARHANDLEOP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *TEMPORARYVARIABLE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *DESTROYTEMPORARYVARIABLE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *VARIABLE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ASSIGN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ASSIGNVARIABLEOP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ASSIGNADD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ASSIGNADDVARIABLEOP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ASSIGNSUB; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ASSIGNSUBVARIABLEOP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *APPLYMOMENTUM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *RESOURCEAPPLYMOMENTUM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SGD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *NOOP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *READVARIABLEOP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *PARALLELCONCATSTART; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *CONSTANTOP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *DEPTHWISECONV2DBACKPROPFILTER; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *DEPTHWISECONV2DBACKPORPINPUT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *DEPTHWISECONV2DFORWARDNATIVE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *DROPOUTGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *APPLYRMSPROPMIXEDPRECISION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *APPLYRMSPROP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *RELU6GRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *AVGPOOLGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *CONCATV2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *CONCATOFFSET; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *LAYERNORMGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *LAYERNORM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *LARS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *DYNAMICSTITCH; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const GATHERV2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const CONVGRADFILTER; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const CONV2D; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const CONV2DBACKPROPINPUT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const FUSEDBATCHNORM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const BIASADDGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ACTIVATIONGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const MAXPOOLWITHARGMAX; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const MAXPOOLGRADWITHARGMAX; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SPARSESOFTMAXCROSSENTROPYWITHLOGITS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SNAPSHOT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const VAR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const MEANGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const TRANSLATE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ADDN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const L2LOSS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const MULTIPLY; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const HUBERLOSSGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const HUBERLOSS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const NEGATIVE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SSDCAST; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SPARSESOFTMAXCROSSENTROPY; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SPARSESOFTMAXCROSSENTROPYGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SSDSQUEEZEFUSION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const CONCATFOUR2FIVE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const CONCATFIVE2FOUR; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SSDREALDIVTILEMUL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SSDSUMMULREALDIVMEAN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const VARIABLEV2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const VARHANDLEOP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const TEMPORARYVARIABLE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const DESTROYTEMPORARYVARIABLE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const VARIABLE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ASSIGN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ASSIGNVARIABLEOP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ASSIGNADD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ASSIGNADDVARIABLEOP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ASSIGNSUB; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ASSIGNSUBVARIABLEOP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const APPLYMOMENTUM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const RESOURCEAPPLYMOMENTUM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SGD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const NOOP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const READVARIABLEOP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const PARALLELCONCATSTART; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const CONSTANTOP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const DEPTHWISECONV2DBACKPROPFILTER; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const DEPTHWISECONV2DBACKPORPINPUT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const DEPTHWISECONV2DFORWARDNATIVE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const DROPOUTGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const APPLYRMSPROPMIXEDPRECISION; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const APPLYRMSPROP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const RELU6GRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const AVGPOOLGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const CONCATV2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const CONCATOFFSET; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const LAYERNORMGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const LAYERNORM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const LARS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const DYNAMICSTITCH; | |||
/***************************************************/ | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SQUARE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *HCOMBROADCAST; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *HCOMALLGATHER; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *HCOMALLREDUCE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *HCOMREDUCESCATTER; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *HCOMSEND; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *HCOMRECEIVE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *HCOMREMOTEREAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *HCOMREMOTEREFREAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *HCOMREMOTEWRITE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *HCOMREMOTESCATTERWRITE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *VARASSIGN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *VARISINITIALIZEDOP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *LogTimeStamp; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ISVARIABLEINITIALIZED; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *STREAMSWITCH; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *STREAMSWITCHN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *STREAMACTIVE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *MEMCPYASYNC; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *MEMCPYADDRASYNC; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *STREAMMERGE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ENDGRAPH; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SEND; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *RECV; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ENDOFSEQUENCE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *LABELSET; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *LABELGOTO; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *LABELGOTOEX; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *LABELSWITCH; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *LABELSWITCHBYINDEX; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ATOMICADDRCLEAN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ABS_GRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ACCUMULATE_N_V2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ACOS_GRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ACOSH_GRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ANY; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *APPROXIMATE_EQUAL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ASIN_GRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ASINH_GRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ATAN_GRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *BROADCAST_TO; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ELU_GRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ADD_V2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *DATAFORMATDIMMAP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *DATAFORMATVECPERMUTE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *BESSELI0E; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *BESSELI1E; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *APPLYADADELTA; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *APPLYADAGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *APPLYADAGRADDA; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *APPLYADAM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *APPLYADAMAX; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *APPLYADDSIGN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *APPLYCENTEREDRMSPROP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *APPLYFTRL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *APPLYFTRLV2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *APPLYGRADIENTDESCENT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *APPLYPOWERSIGN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *APPLYPROXIMALADAGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *APPLYPROXIMALGRADIENTDESCENT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *DEQUANTIZE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *FOCAL_LOSS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *FOCAL_LOSS_GRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SMOOTHL1_LOSS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SMOOTHL1_LOSS_grad; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *REDUCEMEAN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *CONCAT_V2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *ONEHOT_V2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SLICE_V2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *TILE_V2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SUM_V2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SQUARE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const HCOMBROADCAST; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const HCOMALLGATHER; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const HCOMALLREDUCE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const HCOMREDUCESCATTER; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const HCOMSEND; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const HCOMRECEIVE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const HCOMREMOTEREAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const HCOMREMOTEREFREAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const HCOMREMOTEWRITE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const HCOMREMOTESCATTERWRITE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const VARASSIGN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const VARISINITIALIZEDOP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const LogTimeStamp; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ISVARIABLEINITIALIZED; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const STREAMSWITCH; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const STREAMSWITCHN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const STREAMACTIVE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const MEMCPYASYNC; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const MEMCPYADDRASYNC; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const STREAMMERGE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ENDGRAPH; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SEND; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const RECV; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ENDOFSEQUENCE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const LABELSET; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const LABELGOTO; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const LABELGOTOEX; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const LABELSWITCH; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const LABELSWITCHBYINDEX; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ATOMICADDRCLEAN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ABS_GRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ACCUMULATE_N_V2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ACOS_GRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ACOSH_GRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ANY; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const APPROXIMATE_EQUAL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ASIN_GRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ASINH_GRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ATAN_GRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const BROADCAST_TO; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ELU_GRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ADD_V2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const DATAFORMATDIMMAP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const DATAFORMATVECPERMUTE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const BESSELI0E; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const BESSELI1E; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const APPLYADADELTA; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const APPLYADAGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const APPLYADAGRADDA; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const APPLYADAM; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const APPLYADAMAX; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const APPLYADDSIGN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const APPLYCENTEREDRMSPROP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const APPLYFTRL; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const APPLYFTRLV2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const APPLYGRADIENTDESCENT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const APPLYPOWERSIGN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const APPLYPROXIMALADAGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const APPLYPROXIMALGRADIENTDESCENT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const DEQUANTIZE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const FOCAL_LOSS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const FOCAL_LOSS_GRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SMOOTHL1_LOSS; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SMOOTHL1_LOSS_grad; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const REDUCEMEAN; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const CONCAT_V2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const ONEHOT_V2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SLICE_V2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const TILE_V2; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SUM_V2; | |||
// Common type when the operator has the same name | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *DETECTIONOUTPUT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const DETECTIONOUTPUT; | |||
// Custom operator | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *CUSTOMOP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *CUSTOMOP_NCHW; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *CUSTOMOP_NHWC; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *CUSTOMOP_NC1HWC0; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const CUSTOMOP; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const CUSTOMOP_NCHW; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const CUSTOMOP_NHWC; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const CUSTOMOP_NC1HWC0; | |||
// Depthwise 4d_2_6d,6d_2_4d | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *DEPTHWISEWEIGHT4D26D; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *DEPTHWISEWEIGHT6D24D; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const DEPTHWISEWEIGHT4D26D; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const DEPTHWISEWEIGHT6D24D; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SQRTGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *SIGMOIDGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SQRTGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const SIGMOIDGRAD; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *TRANSSHAPE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const TRANSSHAPE; | |||
// Horovod operator | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *HVDCALLBACKALLREDUCE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *HVDCALLBACKALLGATHER; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *HVDCALLBACKBROADCAST; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *HVDWAIT; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const HVDCALLBACKALLREDUCE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const HVDCALLBACKALLGATHER; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const HVDCALLBACKBROADCAST; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const ge::char_t *const HVDWAIT; | |||
/// | |||
/// @brief Magic number of model file | |||
@@ -477,10 +477,10 @@ FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const uint32_t MODEL_FIL | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const uint32_t MODEL_VERSION; ///< Model version 1.0/// | |||
// alpha default value | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const float ALPHA_DEFAULT_VALUE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const float32_t ALPHA_DEFAULT_VALUE; | |||
// beta default value | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const float BETA_DEFAULT_VALUE; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const float32_t BETA_DEFAULT_VALUE; | |||
/// | |||
/// @ingroup domi_omg | |||
@@ -490,7 +490,7 @@ FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const std::string INPUT_ | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const std::string DUMMY_DATA; | |||
// dim default size value | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY static const int32_t DIM_DEFAULT_SIZE = 4; | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY constexpr int32_t DIM_DEFAULT_SIZE = 4; | |||
// for fusion op plugin | |||
FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const std::string ATTR_NAME_FUSIONOP_ORIGINAL_TYPE; | |||
@@ -510,4 +510,4 @@ FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY extern const std::string NODE_N | |||
} // namespace parser | |||
} // namespace ge | |||
#endif // PARSER_COMMON_TYPES_H_ | |||
#endif // INC_FRAMEWORK_OMG_PARSER_TYPES_H_ |
@@ -0,0 +1,31 @@ | |||
/** | |||
* Copyright (c) Huawei Technologies Co., Ltd. 2022. All rights reserved. | |||
* | |||
* Licensed under the Apache License, Version 2.0 (the "License"); | |||
* you may not use this file except in compliance with the License. | |||
* You may obtain a copy of the License at | |||
* | |||
* http://www.apache.org/licenses/LICENSE-2.0 | |||
* | |||
* Unless required by applicable law or agreed to in writing, software | |||
* distributed under the License is distributed on an "AS IS" BASIS, | |||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||
* See the License for the specific language governing permissions and | |||
* limitations under the License. | |||
*/ | |||
#ifndef AIR_INC_FRAMEWORK_PNE_GRAPH_DEPLOYMENT_OPTIMIZER_H_ | |||
#define AIR_INC_FRAMEWORK_PNE_GRAPH_DEPLOYMENT_OPTIMIZER_H_ | |||
#include <map> | |||
#include "framework/common/ge_inner_error_codes.h" | |||
#include "graph/compute_graph.h" | |||
namespace ge { | |||
class GraphDeploymentOptimizer { | |||
public: | |||
static Status OptimizeForAutoDeploy(const ComputeGraphPtr &compute_graph, ComputeGraphPtr &optimized_graph); | |||
}; | |||
} // namespace ge | |||
#endif // AIR_INC_FRAMEWORK_PNE_GRAPH_DEPLOYMENT_OPTIMIZER_H_ |
@@ -69,7 +69,7 @@ class ProcessNodeEngine { | |||
return SUCCESS; | |||
} | |||
virtual Status ParallelPartition(const ComputeGraphPtr &compute_graph) { | |||
virtual Status ParallelPartition(ComputeGraphPtr &compute_graph) { | |||
(void)compute_graph; | |||
return NOT_CHANGED; | |||
} | |||
@@ -22,6 +22,7 @@ namespace memory { | |||
class Block { | |||
public: | |||
// todo use TensorAddress | |||
// todo 需要考虑虚函数对性能的影响 | |||
virtual const void *Addr() const = 0; | |||
virtual void *MutableAddr() = 0; | |||
virtual size_t GetSize() const = 0; | |||
@@ -34,10 +34,10 @@ class VISIBILITY_EXPORT ExeGraphExecutor { | |||
/** | |||
* 设置图执行的输入/输出,需要注意的是,使用者需要自己保证inputs/outputs刷新完全!!! | |||
*/ | |||
ge::graphStatus SpecifyInputs(void **inputs, size_t start, size_t num); | |||
ge::graphStatus SpecifyOutputs(void **outputs, size_t num); | |||
ge::graphStatus Execute(); | |||
ge::graphStatus Execute(ExecutorSubscriber *callback); | |||
ge::graphStatus SpecifyInputs(void *const *inputs, size_t start, size_t num) const; | |||
ge::graphStatus SpecifyOutputs(void *const *outputs, size_t num) const; | |||
ge::graphStatus Execute() const; | |||
ge::graphStatus Execute(ExecutorSubscriber *callback) const; | |||
const void *GetExecutionData() const { | |||
return execution_data_; | |||
@@ -19,6 +19,7 @@ | |||
#include <memory> | |||
#include <vector> | |||
#include <cstdint> | |||
#include <cstdlib> | |||
#include "common/ge_visibility.h" | |||
namespace gert { | |||
@@ -17,22 +17,37 @@ | |||
#ifndef AIR_CXX_INC_FRAMEWORK_RUNTIME_GERT_API_H_ | |||
#define AIR_CXX_INC_FRAMEWORK_RUNTIME_GERT_API_H_ | |||
#include "model_v2_executor.h" | |||
#include "stream_executor.h" | |||
#include "common/ge_types.h" | |||
#include "common/ge_visibility.h" | |||
#include "mem_allocator.h" | |||
namespace gert { | |||
VISIBILITY_EXPORT | |||
std::unique_ptr<ModelV2Executor> LoadExecutorFromFile(const char *model_path, ge::graphStatus &error_code); | |||
std::unique_ptr<ModelV2Executor> LoadExecutorFromFile(const ge::char_t *model_path, ge::graphStatus &error_code); | |||
VISIBILITY_EXPORT | |||
std::unique_ptr<ModelV2Executor> LoadExecutorFromModelData(const ge::ModelData &model_data, | |||
ge::graphStatus &error_code); | |||
VISIBILITY_EXPORT | |||
std::unique_ptr<StreamExecutor> LoadStreamExecutorFromModelData(const ge::ModelData &model_data, const void *weight_ptr, | |||
const size_t weight_size, ge::graphStatus &error_code); | |||
VISIBILITY_EXPORT | |||
ge::graphStatus IsDynamicModel(const void *const model, size_t model_size, bool &is_dynamic_model); | |||
VISIBILITY_EXPORT | |||
ge::graphStatus IsDynamicModel(const char *model_path, bool &is_dynamic_model); | |||
ge::graphStatus IsDynamicModel(const ge::char_t *model_path, bool &is_dynamic_model); | |||
VISIBILITY_EXPORT | |||
std::unique_ptr<ExternalAllocators> CreateSingleOpAllocator(); | |||
ge::graphStatus LoadDataFromFile(const ge::char_t *model_path, ge::ModelData &model_data); | |||
VISIBILITY_EXPORT | |||
std::unique_ptr<ModelV2Executor> LoadExecutorFromModelDataWithMem(const ge::ModelData &model_data, | |||
ge::graphStatus &error_code, const void *weight_ptr, | |||
const size_t weight_size); | |||
VISIBILITY_EXPORT | |||
std::unique_ptr<ExternalAllocators> CreateDefaultAllocator(); | |||
} // namespace gert | |||
#endif // AIR_CXX_INC_FRAMEWORK_RUNTIME_GERT_API_H_ |
@@ -13,8 +13,10 @@ | |||
* See the License for the specific language governing permissions and | |||
* limitations under the License. | |||
*/ | |||
#ifndef AIR_CXX_MEM_ALLOCATOR_H | |||
#define AIR_CXX_MEM_ALLOCATOR_H | |||
#ifndef AIR_MEM_ALLOCATOR_H | |||
#define AIR_MEM_ALLOCATOR_H | |||
#include "exe_graph/runtime/tensor_data.h" | |||
#include "block.h" | |||
#include "exe_graph/runtime/allocator.h" | |||
@@ -24,11 +26,18 @@ struct MemAllocator { | |||
virtual Block *Malloc(size_t size) = 0; | |||
virtual ~MemAllocator() = default; | |||
}; | |||
struct MemSynchronizer { | |||
MemSynchronizer() = default; | |||
virtual ~MemSynchronizer() = default; | |||
// Wait until the memory is actually freed after task completed | |||
virtual void Synchronize() const = 0; | |||
}; | |||
} // namespace memory | |||
struct ExternalAllocators { | |||
public: | |||
memory::MemAllocator *GetAllocator(size_t placement, size_t usage); | |||
memory::MemAllocator *GetAllocator(TensorPlacement placement, size_t usage); | |||
ge::Status SetAllocator(size_t placement, size_t usage, std::unique_ptr<memory::MemAllocator> allocator); | |||
private: | |||
@@ -43,7 +43,7 @@ class VISIBILITY_EXPORT ShapeRange { | |||
class VISIBILITY_EXPORT ModelIoDesc { | |||
public: | |||
const char *GetName() const; | |||
const ge::char_t *GetName() const; | |||
int32_t GetDataType() const; | |||
ge::Format GetStorageFormat() const; | |||
ge::Format GetOriginFormat() const; | |||
@@ -55,7 +55,7 @@ class VISIBILITY_EXPORT ModelIoDesc { | |||
std::vector<std::pair<int64_t, int64_t>> GetOriginShapeRangeVector() const; | |||
std::vector<std::pair<int64_t, int64_t>> GetStorageShapeRangeVector() const; | |||
void SetName(const char *name); | |||
void SetName(const ge::char_t *name); | |||
void SetDataType(int32_t data_type); | |||
void SetStorageFormat(ge::Format format); | |||
void SetOriginFormat(ge::Format format); | |||
@@ -65,7 +65,7 @@ class VISIBILITY_EXPORT ModelIoDesc { | |||
ShapeRange &MutableStorageShapeRange(); | |||
private: | |||
const char *name_; | |||
const ge::char_t *name_; | |||
int32_t data_type_; | |||
StorageFormat format_; | |||
StorageShape shape_; | |||
@@ -100,6 +100,7 @@ class VISIBILITY_EXPORT ModelDesc { | |||
size_t output_num_; | |||
ContinuousVector model_io_descs_; | |||
}; | |||
static_assert(std::is_trivial<ModelDesc>::value, "The class ModelDesc must be a POD"); | |||
} // namespace gert | |||
#endif // AIR_CXX_INC_FRAMEWORK_RUNTIME_MODEL_DESC_H_ |
@@ -27,24 +27,23 @@ | |||
#include "exe_graph_executor.h" | |||
#include "subscriber/executor_subscribers_scheduler.h" | |||
#include "common/ge_types.h" | |||
#include "gert_api.h" | |||
#include "mem_allocator.h" | |||
namespace gert { | |||
enum class ExecutorState { kInit, kLoaded }; | |||
enum SubExeGraphType { kInitExeGraph, kMainExeGraph, kDeInitExeGraph, kSubExeGraphTypeEnd }; | |||
inline const char *GetSubExeGraphTypeStr(SubExeGraphType type) { | |||
constexpr const char *kSubExeGraphTypeStrs[kSubExeGraphTypeEnd] = {"Init", "Main", "DeInit"}; | |||
inline const ge::char_t *GetSubExeGraphTypeStr(const SubExeGraphType type) { | |||
constexpr const ge::char_t *kSubExeGraphTypeStrs[kSubExeGraphTypeEnd] = {"Init", "Main", "DeInit"}; | |||
return kSubExeGraphTypeStrs[type]; | |||
} | |||
enum class ExecuteArgIndex { kExternalAllocator = -2, kStream = -1, kEnd }; | |||
enum class ExecuteArgIndex { kExternalAllocator = -2, kStream, kEnd }; | |||
struct ModelExecuteArg { | |||
rtStream_t stream; | |||
ExternalAllocators *external_allocator; | |||
ModelExecuteArg() : stream(nullptr), external_allocator(nullptr) {} | |||
ModelExecuteArg(rtStream_t stream_, ExternalAllocators *external_allocator_ = nullptr) | |||
ModelExecuteArg(const rtStream_t stream_, ExternalAllocators *const external_allocator_ = nullptr) | |||
: stream(stream_), external_allocator(external_allocator_) {} | |||
}; | |||
static_assert(std::is_standard_layout<ModelExecuteArg>::value, "The class ModelExecuteArg must be a POD"); | |||
@@ -55,6 +54,43 @@ class VISIBILITY_EXPORT ModelV2Executor { | |||
static std::unique_ptr<ModelV2Executor> Create(const ge::ComputeGraphPtr &root_graph); | |||
ge::graphStatus Load(); | |||
/** | |||
* 加载模型,本接口需要在模型执行前被调用。加载流程会完成模型的初始化、将重要数据拷贝到NPU等整个模型生命周期内仅需要执行一次的行为。 | |||
* @param arg | |||
* 模型的执行参数,需要注意的是,此处传入的执行参数应该与Execute接口传入的执行参数具有相同的stream和allocator, | |||
* 否则在load完成后,外部需要调用流同步以保证不出现时序问题 | |||
* @return 成功时返回`ge::GRAPH_SUCCESS` | |||
*/ | |||
ge::graphStatus Load(const ModelExecuteArg &arg); | |||
/** | |||
* 异步执行模型,本接口将模型异步下发到NPU执行,本接口返回不代表模型执行完成,用户需要手动调用流同步等待模型执行完成。 | |||
* 调用本接口前,请确保已经调用`Load`接口 | |||
* | |||
* 用户可以通过多种方式指定输出Tensor,其行为分别为: | |||
* | |||
* * 调用本接口前,用户自行申请了足量空间的输出内存,并通过输出Tensor传入:执行完成后,输出内容被写入到用户申请的输出Tensor。 | |||
* 若用户申请的输出Tensor不够长,那么本接口返回失败。 | |||
* * 用户生成了输出Tensor,但是没有申请输出内存,将不包含输出内存的Tensor传入:本接口内部主动申请输出内存,并将输出内存传出。 | |||
* 若用户没有在arg中指定Allocator,那么本接口输出的内存生命周期与本Executor一致; | |||
* 如果用户在arg中传入了Allocator,那么输出内存将使用用户传入的Allocator申请 | |||
* | |||
* 注意: | |||
* | |||
* 1. 本接口不支持并发调用 | |||
* 2. | |||
* 如果外部指定了Allocator,那么建议Allocator应该与stream绑定,如果出现同一个allocator,匹配不同的stream多次调用Execute接口时, | |||
* 需要满足两个条件:不可以并发调用,在切换stream执行中间,需要对上一条stream做流同步 | |||
* 3. | |||
* 若外部指定了Allocator,在模型执行完成前,不可以将Allocator中的内存归还给操作系统(即使这块内存已经由执行器归还给Allocator) | |||
* | |||
* @param arg 执行参数 | |||
* @param inputs 网络的输入tensor,从调用本接口开始,到流同步等待本模型执行结束之前,用户需要保证传入的Tensor有效 | |||
* @param input_num 输入tensor的数量 | |||
* @param outputs 网络的输出tensor | |||
* @param output_num 输出tensor的数量 | |||
* @return 成功时返回`ge::GRAPH_SUCCESS` | |||
*/ | |||
ge::graphStatus Execute(const ModelExecuteArg &arg, Tensor **inputs, size_t input_num, Tensor **outputs, | |||
size_t output_num); | |||
ge::graphStatus ExecuteSync(Tensor **inputs, size_t input_num, Tensor **outputs, size_t output_num); | |||
@@ -62,7 +98,7 @@ class VISIBILITY_EXPORT ModelV2Executor { | |||
const ModelDesc &GetModelDesc() const; | |||
void SetModelDesc(ModelDesc *model_desc); | |||
ExeGraphExecutor *GetExeGraphExecutor(SubExeGraphType type) { | |||
ExeGraphExecutor *GetExeGraphExecutor(const SubExeGraphType type) { | |||
if (type >= kSubExeGraphTypeEnd) { | |||
return nullptr; | |||
} | |||
@@ -82,8 +118,8 @@ class VISIBILITY_EXPORT ModelV2Executor { | |||
ModelV2Executor(); | |||
private: | |||
std::array<ExeGraphExecutor, kSubExeGraphTypeEnd> graphs_; | |||
ResourceGuard resource_guard_; | |||
std::array<ExeGraphExecutor, kSubExeGraphTypeEnd> graphs_; | |||
ModelDesc *model_desc_ = nullptr; | |||
rtStream_t default_stream_ = nullptr; | |||
ExecutorSubscribersScheduler subscribers_; | |||
@@ -0,0 +1,52 @@ | |||
/** | |||
* Copyright (c) Huawei Technologies Co., Ltd. 2022. All rights reserved. | |||
* | |||
* Licensed under the Apache License, Version 2.0 (the "License"); | |||
* you may not use this file except in compliance with the License. | |||
* You may obtain a copy of the License at | |||
* | |||
* http://www.apache.org/licenses/LICENSE-2.0 | |||
* | |||
* Unless required by applicable law or agreed to in writing, software | |||
* distributed under the License is distributed on an "AS IS" BASIS, | |||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||
* See the License for the specific language governing permissions and | |||
* limitations under the License. | |||
*/ | |||
#ifndef AIR_CXX_STREAM_EXECUTOR_H | |||
#define AIR_CXX_STREAM_EXECUTOR_H | |||
#include <map> | |||
#include <memory> | |||
#include "runtime/base.h" | |||
#include "common/checker.h" | |||
#include "model_v2_executor.h" | |||
namespace gert { | |||
// do not expose the Builder class definition to external api | |||
class ModelV2ExecutorBuilder; | |||
class VISIBILITY_EXPORT StreamExecutor { | |||
public: | |||
explicit StreamExecutor(ModelV2ExecutorBuilder *builder); | |||
StreamExecutor(const StreamExecutor &) = delete; | |||
StreamExecutor &operator=(const StreamExecutor &) = delete; | |||
StreamExecutor(StreamExecutor &&) = delete; | |||
StreamExecutor &operator==(StreamExecutor &&) = delete; | |||
~StreamExecutor(); | |||
ModelV2Executor *GetOrCreateLoaded(rtStream_t stream, const ModelExecuteArg &arg) { | |||
const auto &iter = streams_to_executor_.find(stream); | |||
if (iter != streams_to_executor_.cend()) { | |||
return iter->second.get(); | |||
} | |||
return CreateAndLoad(stream, arg); | |||
} | |||
ge::graphStatus Erase(rtStream_t stream); | |||
private: | |||
ModelV2Executor *CreateAndLoad(rtStream_t stream, const ModelExecuteArg &arg); | |||
private: | |||
ModelV2ExecutorBuilder *builder_; | |||
std::map<rtStream_t, std::unique_ptr<ModelV2Executor>> streams_to_executor_; | |||
}; | |||
} // namespace gert | |||
#endif // AIR_CXX_STREAM_EXECUTOR_H |
@@ -36,17 +36,17 @@ constexpr size_t kInitSubscriberSize = 2UL; | |||
enum class BuiltInSubscriberType { kProfiling, kDumper, kNum }; | |||
enum class ProfilingType { | |||
kCannHost, // 打开Host侧调度的profiling | |||
kDevice, | |||
kGeHost, // 打开GE Host侧调度的profiling | |||
kNum, | |||
kCannHost = 0, // 打开Host侧调度的profiling | |||
kDevice = 1, | |||
kGeHost = 2, // 打开GE Host侧调度的profiling | |||
kNum = 3, | |||
kAll = kNum | |||
}; | |||
static_assert(static_cast<size_t>(ProfilingType::kNum) < sizeof(uint64_t) * 8, | |||
static_assert(static_cast<size_t>(ProfilingType::kNum) < sizeof(uint64_t) * static_cast<size_t>(8), | |||
"The max num of profiling type must less than the width of uint64"); | |||
enum class DumpType { kDataDump, kExceptionDump, kNum, kAll = kNum }; | |||
static_assert(static_cast<size_t>(DumpType::kNum) < sizeof(uint64_t) * 8, | |||
enum class DumpType { kDataDump = 0, kExceptionDump = 1, kNum = 2, kAll = kNum }; | |||
static_assert(static_cast<size_t>(DumpType::kNum) < sizeof(uint64_t) * static_cast<size_t>(8), | |||
"The max num of dumper type must less than the width of uint64"); | |||
class ModelV2Executor; | |||
struct SubscriberExtendInfo { | |||
@@ -70,7 +70,7 @@ class VISIBILITY_EXPORT BuiltInSubscriberUtil { | |||
int>::type = 0> | |||
static uint64_t BuildEnableFlags(const std::vector<T> &enable_types) { | |||
uint64_t flag = 0UL; | |||
for (auto et : enable_types) { | |||
for (const auto &et : enable_types) { | |||
if (et == T::kAll) { | |||
return EnableBit(T::kNum) - 1UL; | |||
} | |||
@@ -21,13 +21,13 @@ | |||
#include "built_in_subscriber_definitions.h" | |||
#include "executor_subscriber_guarder.h" | |||
#include "framework/common/ge_visibility.h" | |||
#include "global_profiling.h" | |||
#include "global_profiler.h" | |||
#include "global_dumper.h" | |||
#include "graph/any_value.h" | |||
namespace gert { | |||
class VISIBILITY_EXPORT ExecutorSubscribersScheduler { | |||
public: | |||
static void OnExecuteEvent(ExecutorSubscribersScheduler *ins, ExecutorEvent event, const void *node, | |||
static void OnExecuteEvent(const ExecutorSubscribersScheduler *ins, ExecutorEvent event, const void *node, | |||
KernelStatus result); | |||
ExecutorSubscribersScheduler() | |||
@@ -35,9 +35,6 @@ class VISIBILITY_EXPORT ExecutorSubscribersScheduler { | |||
built_in_subscribers_ptr_(), | |||
subscribers_(), | |||
subscriber_wrapper_({reinterpret_cast<::SubscriberFunc>(ExecutorSubscribersScheduler::OnExecuteEvent), this}) {} | |||
#ifdef ONLY_COMPILE_OPEN_SRC | |||
~ExecutorSubscribersScheduler(); | |||
#endif | |||
void Init(const SubscriberExtendInfo &extend_info); | |||
ExecutorSubscribersScheduler(const ExecutorSubscribersScheduler &) = delete; | |||
ExecutorSubscribersScheduler &operator=(const ExecutorSubscribersScheduler &) = delete; | |||
@@ -83,7 +80,7 @@ class VISIBILITY_EXPORT ExecutorSubscribersScheduler { | |||
* @param subscriber_type | |||
*/ | |||
void AddBuiltIn(BuiltInSubscriberType subscriber_type, uint64_t enable_flag, const SubscriberExtendInfo &extend_info); | |||
void RemoveSubscriber(void *subscriber_ptr) { | |||
void RemoveSubscriber(const void *subscriber_ptr) { | |||
for (auto iter = subscribers_.begin(); iter != subscribers_.end(); ++iter) { | |||
if (iter->GetSubscriber().arg == subscriber_ptr) { | |||
subscribers_.erase(iter); | |||
@@ -111,8 +108,8 @@ class VISIBILITY_EXPORT ExecutorSubscribersScheduler { | |||
} | |||
bool IsEnable() const { | |||
return enabled_ || GlobalProfilingWrapper::GetInstance()->GetEnableFlags() || | |||
GlobalDumper::GetInstance()->GetEnableFlags(); | |||
return enabled_ || static_cast<bool>(GlobalProfilingWrapper::GetInstance()->GetEnableFlags()) || | |||
static_cast<bool>(GlobalDumper::GetInstance()->GetEnableFlags()); | |||
} | |||
void SetEnable(bool enable_flag) { | |||
enabled_ = enable_flag; | |||
@@ -15,7 +15,6 @@ | |||
*/ | |||
#ifndef AIR_CXX_INC_FRAMEWORK_RUNTIME_EXECUTOR_GLOBAL_DUMPER_H_ | |||
#define AIR_CXX_INC_FRAMEWORK_RUNTIME_EXECUTOR_GLOBAL_DUMPER_H_ | |||
#include <vector> | |||
#include <mutex> | |||
#include "built_in_subscriber_definitions.h" | |||
@@ -23,10 +22,17 @@ | |||
#include "graph/compute_graph.h" | |||
#include "runtime/base.h" | |||
namespace ge { | |||
class ExceptionDumper; | |||
} | |||
namespace gert { | |||
// for global info for exception_dump and global switch | |||
class VISIBILITY_EXPORT GlobalDumper { | |||
public: | |||
GlobalDumper(const GlobalDumper &) = delete; | |||
GlobalDumper(GlobalDumper &&) = delete; | |||
GlobalDumper &operator=(const GlobalDumper &) = delete; | |||
GlobalDumper &operator==(GlobalDumper &&) = delete; | |||
static GlobalDumper *GetInstance() { | |||
static GlobalDumper global_dumper; | |||
return &global_dumper; | |||
@@ -34,14 +40,9 @@ class VISIBILITY_EXPORT GlobalDumper { | |||
static void OnGlobalDumperSwitch(void *ins, uint64_t enable_flags); | |||
void AddExceptionInfo(const rtExceptionInfo *exception_data) { | |||
const std::lock_guard<std::mutex> lk(exception_infos_mutex_); | |||
exception_infos_.emplace_back(*exception_data); | |||
} | |||
void AddExceptionInfo(const rtExceptionInfo *exception_data) const; | |||
const std::vector<rtExceptionInfo> &GetExceptionInfos() const { | |||
return exception_infos_; | |||
} | |||
ge::ExceptionDumper *MutableExceptionDumper() const; | |||
void SetEnableFlags(const uint64_t enable_flags) { | |||
enable_flags_ = enable_flags; | |||
@@ -52,20 +53,12 @@ class VISIBILITY_EXPORT GlobalDumper { | |||
}; | |||
bool IsEnable(DumpType dump_type) const { | |||
return enable_flags_ & BuiltInSubscriberUtil::EnableBit<DumpType>(dump_type); | |||
return static_cast<bool>(enable_flags_ & BuiltInSubscriberUtil::EnableBit<DumpType>(dump_type)); | |||
} | |||
private: | |||
GlobalDumper(); | |||
GlobalDumper(const GlobalDumper &) = delete; | |||
GlobalDumper(GlobalDumper &&) = delete; | |||
GlobalDumper &operator=(const GlobalDumper &) = delete; | |||
GlobalDumper &operator==(GlobalDumper &&) = delete; | |||
private: | |||
uint64_t enable_flags_{0UL}; | |||
std::vector<rtExceptionInfo> exception_infos_; | |||
std::mutex exception_infos_mutex_; | |||
}; | |||
} // namespace gert | |||
#endif |
@@ -13,8 +13,8 @@ | |||
* See the License for the specific language governing permissions and | |||
* limitations under the License. | |||
*/ | |||
#ifndef AIR_CXX_INC_FRAMEWORK_RUNTIME_SUBSCRIBER_GLOBAL_PROFILING_H_ | |||
#define AIR_CXX_INC_FRAMEWORK_RUNTIME_SUBSCRIBER_GLOBAL_PROFILING_H_ | |||
#ifndef AIR_CXX_INC_FRAMEWORK_RUNTIME_SUBSCRIBER_GLOBAL_PROFILER_H_ | |||
#define AIR_CXX_INC_FRAMEWORK_RUNTIME_SUBSCRIBER_GLOBAL_PROFILER_H_ | |||
#include <algorithm> | |||
#include <memory> | |||
@@ -77,7 +77,7 @@ class VISIBILITY_EXPORT GlobalProfilingWrapper { | |||
return global_profiler_.get(); | |||
} | |||
void SetEnableFlags(uint64_t enable_flags) { | |||
void SetEnableFlags(const uint64_t enable_flags) { | |||
enable_flags_ = enable_flags; | |||
} | |||
@@ -1 +1 @@ | |||
Subproject commit dd39e0ff029fc32d7cfff0147824e7c302d7796a | |||
Subproject commit b0adaafa9183a8e453ebcd23710ca9c61159c51f |
@@ -1,23 +1,8 @@ | |||
/** | |||
* Copyright 2019-2020 Huawei Technologies Co., Ltd | |||
* | |||
* Licensed under the Apache License, Version 2.0 (the "License"); | |||
* you may not use this file except in compliance with the License. | |||
* You may obtain a copy of the License at | |||
* | |||
* http://www.apache.org/licenses/LICENSE-2.0 | |||
* | |||
* Unless required by applicable law or agreed to in writing, software | |||
* distributed under the License is distributed on an "AS IS" BASIS, | |||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |||
* See the License for the specific language governing permissions and | |||
* limitations under the License. | |||
*/ | |||
/** | |||
* @file base.h | |||
* @brief HCOM data type definition | |||
* | |||
/* | |||
* Copyright (c) Huawei Technologies Co., Ltd. 2019-2022. All rights reserved. | |||
* Description: HCOM data type definition | |||
* Author: ligang | |||
* Create: 2019-05-24 | |||
*/ | |||
#ifndef HCCL_BASE_H_ | |||
@@ -139,22 +124,12 @@ typedef void *rtModel_t; | |||
struct HcomOperation { | |||
std::string hcclType; | |||
void *inputPtr; | |||
void *outputPtr; | |||
u64 count; | |||
HcclDataType dataType; | |||
HcclReduceOp opType; | |||
u32 root; | |||
HcomOperation() | |||
{ | |||
inputPtr = nullptr; | |||
outputPtr = nullptr; | |||
count = 0; | |||
dataType = HCCL_DATA_TYPE_RESERVED; | |||
opType = HCCL_REDUCE_RESERVED; | |||
root = 0; | |||
} | |||
void *inputPtr{nullptr}; | |||
void *outputPtr{nullptr}; | |||
u64 count{0}; | |||
HcclDataType dataType{HCCL_DATA_TYPE_RESERVED}; | |||
HcclReduceOp opType{HCCL_REDUCE_RESERVED}; | |||
u32 root{0}; | |||
}; | |||
struct HcomRemoteAccessAddrInfo { | |||
@@ -165,15 +140,24 @@ struct HcomRemoteAccessAddrInfo { | |||
}; | |||
struct HcomAllToAllVParams { | |||
void *sendbuf; // device mem | |||
void *sendcounts; // device mem; Type: uint_64 | |||
void *sdispls; // device mem; Type: uint_64 | |||
HcclDataType sendtype; | |||
void *recvbuf; // device mem | |||
void *recvcounts; // device mem; Type: uint_64 | |||
void *rdispls; // device mem; Type: uint_64 | |||
HcclDataType recvtype; | |||
const char *group; // not used now | |||
void *sendbuf{nullptr}; // device mem | |||
void *sendcounts{nullptr}; // device mem; Type: uint_64 | |||
void *sdispls{nullptr}; // device mem; Type: uint_64 | |||
HcclDataType sendtype{HCCL_DATA_TYPE_RESERVED}; | |||
void *recvbuf{nullptr}; // device mem | |||
void *recvcounts{nullptr}; // device mem; Type: uint_64 | |||
void *rdispls{nullptr}; // device mem; Type: uint_64 | |||
HcclDataType recvtype{HCCL_DATA_TYPE_RESERVED}; | |||
const char *group{nullptr}; // not used now | |||
}; | |||
struct HcomAllToAllVCParams { | |||
void *sendbuf{nullptr}; // device mem | |||
HcclDataType sendtype{HCCL_DATA_TYPE_RESERVED}; | |||
void *recvbuf{nullptr}; // device mem | |||
HcclDataType recvtype{HCCL_DATA_TYPE_RESERVED}; | |||
void *sendcountmatrix{nullptr}; // device mem; Type: uint_64 | |||
const char *group{nullptr}; // not used now | |||
}; | |||
struct HcomGatherAllToAllVParams { | |||
@@ -1,60 +1,51 @@ | |||
approvers: | |||
- gegenhua | |||
- zhoujian25 | |||
- qiaohairong | |||
reviewers: | |||
- chuqingxi | |||
- wang-jintang | |||
- luanma_bl | |||
- chen-kang30 | |||
- li-xulong | |||
- Allan_Yu | |||
- minshen | |||
- pan-jixing | |||
- yl_wang | |||
- lijie176 | |||
- mabing726 | |||
- miao-fangzheng | |||
- huang-qiang002 | |||
- su-yueming | |||
- chenpeng-hw | |||
- wang_jianle | |||
- luanma_bl | |||
- LDLD0524 | |||
- wywismygod2020 | |||
- koala-zhang | |||
- zhu-jingjing | |||
- zhaozhihui5 | |||
- simbaliuxx | |||
- lyxyz | |||
- zhou-qilong | |||
- block0219 | |||
- hanfuwei | |||
- xchu42 | |||
- sheng-nan | |||
- yangjing88 | |||
- alexlak | |||
- xig514 | |||
- jellylj | |||
- brightlyking | |||
- chengsheng304064 | |||
- liuzhenyuhw | |||
- djh602 | |||
- wangjiangben_hw | |||
- li1jie | |||
- liujun2014 | |||
- soupkey | |||
- wu-shengji | |||
- cimeng | |||
- granpad7 | |||
- tc1qaz | |||
- Ronnie_zheng | |||
- xiexianhu | |||
- zhaoping12 | |||
- tanshengshun | |||
- fanqirui | |||
- xu-binglin | |||
- yangyang016 | |||
- zhangzhongzt | |||
- gegenhua | |||
- qiaohairong | |||
- wu-mingli | |||
- fang-guocan | |||
- wangtao43 | |||
- changhaixun | |||
- yan-cailiang | |||
- jiangtao_vector | |||
- shunqi | |||
- wangjun-qt | |||
- brightlyking | |||
- wangchao285 | |||
- wu-jinfu | |||
- llim | |||
- Wang_Xing001 | |||
- liuyi66 | |||
- fan-zhaoyun | |||
- tangqunzhang | |||
- chy_huawei | |||
- ly-qianxiao | |||
- wangzhendong07 | |||
- crystalhu | |||
- lin-wei32 | |||
- wangjiangben_hw | |||
- clinglai | |||
options: | |||
no_parent_owners: true |
@@ -65,6 +65,7 @@ | |||
#include "resource_variable_ops.h" | |||
#include "rnn.h" | |||
#include "rpn_ops.h" | |||
#include "randomdsa_ops.h" | |||
#include "save_ops.h" | |||
#include "selection_ops.h" | |||
#include "set_ops.h" | |||
@@ -2576,5 +2576,177 @@ REG_OP(FlowFunc) | |||
.ATTR(output_shapes, ListListInt, {}) | |||
.REQUIRED_ATTR(output_types, ListType) | |||
.OP_END_FACTORY_REG(FlowFunc) | |||
/** | |||
* @brief ouputs a tensor copy from the tensor at 'position' in input_sequence. \n | |||
* @par Inputs: | |||
* @li handle: the handle of sequence. | |||
* @li index: position of the tensor in the sequence. negative value means | |||
* counting position from back, accepted range in [-n, n - 1], | |||
* where n is the number of tensors in sequence, | |||
* it must be a scalar(tensor of empty shape), it is scalar. \n | |||
* @par Outputs: | |||
* @li y: output tensor at the specified position in the input sequence. \n | |||
*/ | |||
REG_OP(SequenceAt) | |||
.INPUT(handle, TensorType({DT_RESOURCE})) | |||
.INPUT(index, TensorType({DT_INT32, DT_INT64})) | |||
.OUTPUT(y, TensorType({DT_UINT8, DT_UINT16, DT_UINT32, DT_UINT64, DT_INT8, DT_INT16, \ | |||
DT_INT32, DT_INT64, DT_FLOAT16, DT_FLOAT, DT_DOUBLE, DT_BOOL, DT_COMPLEX64, \ | |||
DT_COMPLEX128})) | |||
.OP_END_FACTORY_REG(SequenceAt) | |||
/** | |||
* @brief constrct a tensor sequence cotaining 'input' tensors, | |||
* all tensors in 'inputs' must have the same data type. \n | |||
* @par Inputs: | |||
* @li inputs: A list of input tensor objects. It's a dynamic input. \n | |||
* @par Outputs: | |||
* @li handle: Sequence enclosing the input tensors. \n | |||
*/ | |||
REG_OP(SequenceConstruct) | |||
.DYNAMIC_INPUT(inputs, TensorType({DT_UINT8, DT_UINT16, DT_UINT32, DT_UINT64, DT_INT8, \ | |||
DT_INT16, DT_INT32, DT_INT64, DT_FLOAT16, DT_FLOAT, DT_DOUBLE, DT_BOOL, DT_COMPLEX64, \ | |||
DT_COMPLEX128})) | |||
.OUTPUT(handle, TensorType({DT_RESOURCE})) | |||
.OP_END_FACTORY_REG(SequenceConstruct) | |||
/** | |||
* @brief construct an empty tensor sequence, with given data type. \n | |||
* @par Outputs: | |||
* @li handle: empty sequence. \n | |||
* @par Attributes: | |||
* @li dtype: the data type of the tensors in the output sequence, | |||
* the default value is float. \n | |||
*/ | |||
REG_OP(SequenceEmpty) | |||
.OUTPUT(handle, TensorType({DT_RESOURCE})) | |||
.ATTR(dtype, Type, DT_FLOAT) | |||
.OP_END_FACTORY_REG(SequenceEmpty) | |||
/** | |||
* @brief ouputs a tensor sequence that remove the tensor at 'position' from input_sequence. \n | |||
* @par Inputs: | |||
* @li handle: the handle of sequence. | |||
* @li index: position of the tensor in the sequence. negative value means | |||
* counting position from back, accepted range in [-n, n - 1], | |||
* where n is the number of tensors in sequence, | |||
* it must be a scalar(tensor of empty shape), it is scalar. \n | |||
* @par Outputs: | |||
* @li handle_y: the handle of the sequence that has the tensor | |||
* at the specified position removed. \n | |||
*/ | |||
REG_OP(SequenceErase) | |||
.INPUT(handle, TensorType({DT_RESOURCE})) | |||
.OPTIONAL_INPUT(index, TensorType({DT_INT32, DT_INT64})) | |||
.OUTPUT(handle_y, TensorType({DT_RESOURCE})) | |||
.OP_END_FACTORY_REG(SequenceErase) | |||
/** | |||
* @brief ouputs a tensor sequence that insert tensor into sequence at 'position', | |||
* tensor must have the same data type as input_sequence. \n | |||
* @par Inputs: | |||
* @li handle: the handle of sequence. | |||
* @li value: tensor to be inserted into the input sequence. \n | |||
* @li index: position of the tensor in the sequence. negative value means | |||
* counting position from back, accepted range in [-n, n - 1], | |||
* where n is the number of tensors in sequence, | |||
* it must be a scalar(tensor of empty shape), it is scalar. \n | |||
* @par Outputs: | |||
* @li handle_y: output sequence that contains the inserted tensor | |||
* at the given positon. \n | |||
*/ | |||
REG_OP(SequenceInsert) | |||
.INPUT(handle, TensorType({DT_RESOURCE})) | |||
.INPUT(value, TensorType({DT_UINT8, DT_UINT16, DT_UINT32, DT_UINT64, DT_INT8, \ | |||
DT_INT16, DT_INT32, DT_INT64, DT_FLOAT16, DT_FLOAT, DT_DOUBLE, DT_BOOL, DT_COMPLEX64, \ | |||
DT_COMPLEX128})) | |||
.OPTIONAL_INPUT(index, TensorType({DT_INT32, DT_INT64})) | |||
.OUTPUT(handle_y, TensorType({DT_RESOURCE})) | |||
.OP_END_FACTORY_REG(SequenceInsert) | |||
/** | |||
* @brief produces a scalar(tensor of empty shape) containing the number | |||
* of tensors in input_sequence. \n | |||
* @par Inputs: | |||
* @li handle: the handle of sequence. \n | |||
* @par Outputs: | |||
* @li length: length of input sequence, it must be a scalar (tensor of empty shape) \n | |||
*/ | |||
REG_OP(SequenceLength) | |||
.INPUT(handle, TensorType({DT_RESOURCE})) | |||
.OUTPUT(length, TensorType({DT_INT64})) | |||
.OP_END_FACTORY_REG(SequenceLength) | |||
/** | |||
* @brief split a tensor into a sequence of tensors, along the specified axis, | |||
* length of the parts can be specified using argument 'split'. \n | |||
* @par Inputs: | |||
* @li x: the tensor to split. \n | |||
* @li split: length of each output, it cat be either a scalar or 1-D tensor, | |||
* all value must be >= 0. if split is a scalar, then input will be split into | |||
* equally sized chunks, last chunk will be smaller if input size along the given | |||
* axis is not divisible by split, otherwise the tensor is split into size(split) | |||
* chunks, with lengths of the parts on axis specified in split. in the scenario, | |||
* the sum of entries in split must be equal to the dimision size of input tensor | |||
* on axis \n | |||
* @par Outputs: | |||
* @li handle: one or more outputs forming a sequence of tensor after spliting. \n | |||
* @par Attributes: | |||
* @li axis: which axis to split on, a negative value means counting dimensions from | |||
* the back, accepted range is [-rank, rank - 1]. | |||
* @li keep_dims: keep the split dimension or not, default 1,which means we keep split | |||
* dimension,if input 'split' is specified, this attribute is ignored. | |||
*/ | |||
REG_OP(SplitToSequence) | |||
.INPUT(x, TensorType({DT_UINT8, DT_UINT16, DT_UINT32, DT_UINT64, DT_INT8, \ | |||
DT_INT16, DT_INT32, DT_INT64, DT_FLOAT16, DT_FLOAT, DT_DOUBLE, DT_BOOL, DT_COMPLEX64, \ | |||
DT_COMPLEX128})) | |||
.OPTIONAL_INPUT(split, TensorType({DT_INT32, DT_INT64})) | |||
.OUTPUT(handle, TensorType({DT_RESOURCE})) | |||
.ATTR(axis, Int, 0) | |||
.ATTR(keepdims, Bool, true) | |||
.OP_END_FACTORY_REG(SplitToSequence) | |||
/** | |||
* @brief concatenate a sequence of tensors into a single tensor, all input tensors must | |||
* have the same shape, except for the dimension size of the axis to concatenate on. by default | |||
* new_axis is 0, the behavior is similar to numpy.concatenate. when new_axis is 1, the behavior | |||
* is similar to numpy.stack. \n | |||
* @par Inputs: | |||
* @li handle: sequence of tensors for concatenation. \n | |||
* @par Outputs: | |||
* @li y: concatenated tensor. \n | |||
* @par Attributes: | |||
* @li axis: which axis to concat on, accepted range in [-r, r - 1], where r is the rank of input | |||
* tensor, when new_axis is 1,accepted range is [-r - 1, r] | |||
* @li new_axis: insert and concatnate on a new axis or not. default 0 means do not insert new axis. | |||
*/ | |||
REG_OP(ConcatFromSequence) | |||
.INPUT(handle, TensorType({DT_RESOURCE})) | |||
.OUTPUT(y, TensorType({DT_UINT8, DT_UINT16, DT_UINT32, DT_UINT64, DT_INT8, \ | |||
DT_INT16, DT_INT32, DT_INT64, DT_FLOAT16, DT_FLOAT, DT_DOUBLE, DT_BOOL, DT_COMPLEX64, \ | |||
DT_COMPLEX128})) | |||
.REQUIRED_ATTR(axis, Int) | |||
.ATTR(new_axis, Int, 0) | |||
.OP_END_FACTORY_REG(ConcatFromSequence) | |||
} // namespace ge | |||
#endif // OPS_BUILT_IN_OP_PROTO_INC_DATA_FLOW_OPS_H_ |
@@ -2944,9 +2944,9 @@ REG_OP(AdamApplyOne) | |||
*@par Outputs: | |||
*Three outputs, including: | |||
* @li output0: A Tensor. Must be one of the following types: float16, float32. | |||
* @li output1: A Tensor. Must be one of the following types: float16, float32. | |||
* @li output2: A Tensor. Must be one of the following types: float16, float32. \n | |||
* @li input1: A Tensor. Must be one of the following types: float16, float32. | |||
* @li input2: A Tensor. Must be one of the following types: float16, float32. | |||
* @li input3: A Tensor. Must be one of the following types: float16, float32. \n | |||
*@par Restrictions: | |||
*Warning: THIS FUNCTION IS EXPERIMENTAL. Please do not use. | |||
@@ -2963,9 +2963,9 @@ REG_OP(AdamApplyOneWithDecayAssign) | |||
.INPUT(mul3_x, TensorType({DT_FLOAT16,DT_FLOAT})) | |||
.INPUT(mul4_x, TensorType({DT_FLOAT16,DT_FLOAT})) | |||
.INPUT(add2_y, TensorType({DT_FLOAT16,DT_FLOAT})) | |||
.OUTPUT(output0, TensorType({DT_FLOAT16,DT_FLOAT})) | |||
.OUTPUT(output1, TensorType({DT_FLOAT16,DT_FLOAT})) | |||
.OUTPUT(output2, TensorType({DT_FLOAT16,DT_FLOAT})) | |||
.OUTPUT(input1, TensorType({DT_FLOAT16,DT_FLOAT})) | |||
.OUTPUT(input2, TensorType({DT_FLOAT16,DT_FLOAT})) | |||
.OUTPUT(input3, TensorType({DT_FLOAT16,DT_FLOAT})) | |||
.OP_END_FACTORY_REG(AdamApplyOneWithDecayAssign) | |||
/** | |||
@@ -2986,9 +2986,9 @@ REG_OP(AdamApplyOneWithDecayAssign) | |||
*@par Outputs: | |||
*Three outputs, including: | |||
* @li output0: A Tensor. Must be one of the following types: float16, float32. | |||
* @li output1: A Tensor. Must be one of the following types: float16, float32. | |||
* @li output2: A Tensor. Must be one of the following types: float16, float32. \n | |||
* @li input1: A Tensor. Must be one of the following types: float16, float32. | |||
* @li input2: A Tensor. Must be one of the following types: float16, float32. | |||
* @li input3: A Tensor. Must be one of the following types: float16, float32. \n | |||
*@par Restrictions: | |||
*Warning: THIS FUNCTION IS EXPERIMENTAL. Please do not use. | |||
@@ -3004,9 +3004,9 @@ REG_OP(AdamApplyOneAssign) | |||
.INPUT(mul2_x, TensorType({DT_FLOAT16,DT_FLOAT})) | |||
.INPUT(mul3_x, TensorType({DT_FLOAT16,DT_FLOAT})) | |||
.INPUT(add2_y, TensorType({DT_FLOAT16,DT_FLOAT})) | |||
.OUTPUT(output0, TensorType({DT_FLOAT16,DT_FLOAT})) | |||
.OUTPUT(output1, TensorType({DT_FLOAT16,DT_FLOAT})) | |||
.OUTPUT(output2, TensorType({DT_FLOAT16,DT_FLOAT})) | |||
.OUTPUT(input1, TensorType({DT_FLOAT16,DT_FLOAT})) | |||
.OUTPUT(input2, TensorType({DT_FLOAT16,DT_FLOAT})) | |||
.OUTPUT(input3, TensorType({DT_FLOAT16,DT_FLOAT})) | |||
.OP_END_FACTORY_REG(AdamApplyOneAssign) | |||
/** | |||
@@ -3383,17 +3383,20 @@ REG_OP(Muls) | |||
*@par Inputs: | |||
*One input, including: | |||
*x: A Tensor. Must be one of the following types:float32, float16, int64, int32, int16, bool. | |||
*x: A Tensor. Must be one of the following types:double, float32, float16, int64, int32, int16, bool. | |||
*@par Outputs: | |||
*y: A Tensor. Has the same type and shape as "x1". \n | |||
*@par Attributes: | |||
*value: A scale. Must be float. \n | |||
*@par Third-party framework compatibility: | |||
* Compatible with the Pytorch operator fills. | |||
*/ | |||
REG_OP(Fills) | |||
.INPUT(x, TensorType({DT_FLOAT, DT_FLOAT16, DT_INT64, DT_INT32, DT_INT16, DT_BOOL})) | |||
.OUTPUT(y, TensorType({DT_FLOAT, DT_FLOAT16, DT_INT64, DT_INT32, DT_INT16, DT_BOOL})) | |||
.INPUT(x, TensorType({DT_DOUBLE, DT_FLOAT, DT_FLOAT16, DT_INT64, DT_INT32, DT_INT16, DT_BOOL})) | |||
.OUTPUT(y, TensorType({DT_DOUBLE, DT_FLOAT, DT_FLOAT16, DT_INT64, DT_INT32, DT_INT16, DT_BOOL})) | |||
.REQUIRED_ATTR(value, Float) | |||
.OP_END_FACTORY_REG(Fills) | |||
@@ -4032,6 +4035,31 @@ REG_OP(Dawsn) | |||
.INPUT(x, TensorType({DT_BF16, DT_FLOAT16, DT_FLOAT, DT_DOUBLE})) | |||
.OUTPUT(y, TensorType({DT_BF16, DT_FLOAT16, DT_FLOAT, DT_DOUBLE})) | |||
.OP_END_FACTORY_REG(Dawsn) | |||
/** | |||
* @brief The operator casts the elements of a given input tensor (the first input) | |||
* to the same data type as the elements of the second input tensor. \n | |||
* @par Inputs: | |||
* Two inputs, including: | |||
* @li x:A Tensor. Must be one of the following types: | |||
float16, float, int8, int32, uint32, uint8, int64, uint64, int16, uint16, double, | |||
complex64, complex128, qint8, quint8, qint16, quint16, qint32, bfloat16, bool, string. \n | |||
* @li target:A Tensor. Must be one of the following types: | |||
float16, float, int8, int32, uint32, uint8, int64, uint64, int16, uint16, double, | |||
complex64, complex128, qint8, quint8, qint16, quint16, qint32, bfloat16, bool, string. \n | |||
* @par Outputs: | |||
* y:A Tensor with same shape as x, and data type is specified by target. | |||
* @par Third-party framework compatibility | |||
* Compatible with the onnx operator CastLike. | |||
*/ | |||
REG_OP(CastLike) | |||
.INPUT(x, TensorType({BasicType(), DT_BOOL, DT_STRING})) | |||
.INPUT(target, TensorType({BasicType(), DT_BOOL, DT_STRING})) | |||
.OUTPUT(y, TensorType({BasicType(), DT_BOOL, DT_STRING})) | |||
.OP_END_FACTORY_REG(CastLike) | |||
} // namespace ge | |||
#endif // OPS_BUILT_IN_OP_PROTO_INC_ELEWISE_CALCULATION_OPS_H_ |
@@ -219,6 +219,40 @@ REG_OP(Index) | |||
.OP_END_FACTORY_REG(Index) | |||
/** | |||
* @brief According to the index number of indexes, replace the value | |||
* corresponding to X with the value. | |||
* @par Inputs: | |||
* Five inputs, including: | |||
* @li x: A ND Tensor. | |||
* @li value: A Tensor of the same type as "x". | |||
* @li indexed_sizes: A 1D Tensor of int64 with shape (N). Sizes for each one of the indexed data. | |||
* @li indexed_strides: A 1D Tensor of int64 with shape (N). Strides for each one of the indexed data. | |||
* @li indices: Dynamic input. A Tensor of the indices. | |||
* @par Attributes: | |||
* @li accumulate: Does it support self accumulation. Defaults to false. | |||
* @par Outputs: | |||
* @li x: A Tensor. | |||
* @par Third-party framework compatibility | |||
* Compatible with the Pytorch operator index_put. | |||
* @par Restrictions: | |||
* Warning:THIS FUNCTION IS EXPERIMENTAL. Please do not use. | |||
*/ | |||
REG_OP(IndexPutV2) | |||
.INPUT(x, TensorType::BasicType()) | |||
.INPUT(value, TensorType::BasicType()) | |||
.INPUT(indexed_sizes, TensorType({DT_INT64})) | |||
.INPUT(indexed_strides, TensorType({DT_INT64})) | |||
.DYNAMIC_INPUT(indices, TensorType({DT_INT64})) | |||
.OUTPUT(x, TensorType::BasicType()) | |||
.ATTR(accumulate, Bool, false) | |||
.OP_END_FACTORY_REG(IndexPutV2) | |||
/** | |||
* @brief Performs average pooling on the input. Used in the combination of conv + avgpoolupdate to replace avgpool | |||
* @par Inputs: | |||
* x1: Output of upstream Conv2d. A tensor of type float16, float32. | |||
@@ -374,6 +408,86 @@ REG_OP(DecodeJpegPre) | |||
.REQUIRED_ATTR(w_range, ListInt) | |||
.REQUIRED_ATTR(h_range, ListInt) | |||
.OP_END_FACTORY_REG(DecodeJpegPre) | |||
/** | |||
* @brief init PartitionMap table. \n | |||
* @par Inputs: | |||
* @li ps_num: A Tensor, dtype is uint32. 0-D. indicates ps number. | |||
* @li ps_ids: A Tensor, dtype is uint32. 1-D. indicates the id of ps. \n | |||
* @par Attributes: | |||
* @li partition_num: A Int, indicates the number of partition. \n | |||
*/ | |||
REG_OP(InitPartitionMap) | |||
.INPUT(ps_num, TensorType({DT_UINT32})) | |||
.INPUT(ps_ids, TensorType({DT_UINT32})) | |||
.ATTR(partition_num, Int, 65537) | |||
.OP_END_FACTORY_REG(InitPartitionMap) | |||
/** | |||
* @brief uninit PartitionMap table. \n | |||
*/ | |||
REG_OP(UninitPartitionMap) | |||
.OP_END_FACTORY_REG(UninitPartitionMap) | |||
/** | |||
* @brief init Embedding hashtable. \n | |||
* @par Inputs: | |||
* @li table_id: A Tensor, dtype is uint32. 0-D. indicates the id of hashtable. \n | |||
* @par Attributes: | |||
* @li bucket_size: A Int. \n | |||
*/ | |||
REG_OP(InitEmbeddingHashmap) | |||
.INPUT(table_id, TensorType({DT_UINT32})) | |||
.ATTR(bucket_size, Int, 0) | |||
.OP_END_FACTORY_REG(InitEmbeddingHashmap) | |||
/** | |||
* @brief embedding hsahtable data import. \n | |||
* @par Inputs: | |||
* @li file_path: A Tensor, dtype is string. 0-D. indicates embedding filepath. | |||
* @li file_name: A Tensor, dtype is string. 0-D. indicates embedding filename. | |||
* @li ps_id: A Tensor, dtype is uint32. 0-D. indicates the id of ps. | |||
* @li table_id: A Tensor, dtype is uint32. 0-D. indicates the id of hashtable. | |||
* @li embedding_dim: A Tensor, dtype is uint32. 0-D. indicates the hashtable value number. \n | |||
*/ | |||
REG_OP(EmbeddingTableImport) | |||
.INPUT(file_path, TensorType({DT_STRING})) | |||
.INPUT(file_name, TensorType({DT_STRING})) | |||
.INPUT(ps_id, TensorType({DT_UINT32})) | |||
.INPUT(table_id, TensorType({DT_UINT32})) | |||
.INPUT(embedding_dim, TensorType({DT_UINT32})) | |||
.OP_END_FACTORY_REG(EmbeddingTableImport) | |||
/** | |||
* @brief embedding hsahtable data lookup. \n | |||
* @par Inputs: | |||
* @li table_id: A Tensor, dtype is uint32. 0-D. indicates the id of hashtable. | |||
* @li keys: A Tensor, dtype is uint32. 1-D. indicates the hashtable key. \n | |||
* @par Outputs: | |||
* @li values: indicates the hashtable value. \n | |||
*/ | |||
REG_OP(EmbeddingTableFind) | |||
.INPUT(table_id, TensorType({DT_UINT32})) | |||
.INPUT(keys, TensorType({DT_UINT64})) | |||
.OUTPUT(values, TensorType({DT_FLOAT})) | |||
.OP_END_FACTORY_REG(EmbeddingTableFind) | |||
/** | |||
* @brief uninit embedding hsahtable. \n | |||
* @par Inputs: | |||
* @li table_id: A Tensor, dtype is uint32. 0-D. indicates the id of hashtable. \n | |||
*/ | |||
REG_OP(UninitEmbeddingHashmap) | |||
.INPUT(table_id, TensorType({DT_UINT32})) | |||
.OP_END_FACTORY_REG(UninitEmbeddingHashmap) | |||
} // namespace ge | |||
#endif // OPS_BUILT_IN_OP_PROTO_INC_EXPERIMENT_OPS_H_ |
@@ -167,6 +167,56 @@ REG_OP(CropAndResize) | |||
.OP_END_FACTORY_REG(CropAndResize) | |||
/** | |||
* @brief Extracts crops from the input image tensor and resizes them. Extracts | |||
crops from the input image tensor and resizes them using bilinear sampling or | |||
nearest neighbor sampling to a common output size specified by crop_size . \n | |||
* @par Inputs: | |||
* Input x must be a 4-D tensor. Inputs include: | |||
* @li x: A Tensor. Must be one of the following types:uint8, uint16, int8, | |||
int16, int32, int64, float16, float, double. A 4-D tensor of shape | |||
[batch, image_height, image_width, depth]. The format must be NHWC. | |||
* @li boxes: A Tensor. Must be the float types. | |||
A 2-D tensor of shape [num_boxes, 4]. | |||
* @li box_index: A Tensor of type int32. A 1-D tensor of shape [num_boxes] with | |||
int32 values in [0, batch). | |||
* @li crop_size: A Tensor of type int32. A 1-D tensor of 2 elements, | |||
crop_size = [crop_height, crop_width]. | |||
All cropped image patches are resized to this size . \n | |||
* @par Attributes: | |||
* @li extrapolation_value: An optional float. Defaults to 0. Value used for | |||
extrapolation, when applicable. | |||
* @li method: An optional string from: '"bilinear", "nearest"'. Defaults to | |||
"bilinear". Currently two sampling methods are supported: Bilinear | |||
and NearestNeighbor . | |||
* @li dtype: An Type attr, support type list [uint8, float16, float]. | |||
Defaults to DT_FLOAT . \n | |||
* @par Outputs: | |||
* y: A Tensor. Must be one of the following types: uint8, float16, float. | |||
The format must be NHWC. \n | |||
* @attention Constraints: | |||
* Input images must be a 4-D tensor . \n | |||
* @par Third-party framework compatibility | |||
* Compatible with tensorflow CropAndResize operator. | |||
*/ | |||
REG_OP(CropAndResizeV2) | |||
.INPUT(x, TensorType({DT_UINT8, DT_UINT16, DT_INT8, \ | |||
DT_INT16, DT_INT32, DT_INT64, DT_FLOAT16, DT_FLOAT, DT_DOUBLE})) | |||
.INPUT(boxes, TensorType({DT_FLOAT})) | |||
.INPUT(box_index, TensorType({DT_INT32})) | |||
.INPUT(crop_size, TensorType({DT_INT32})) | |||
.OUTPUT(y, TensorType({DT_UINT8, DT_FLOAT16, DT_FLOAT})) | |||
.ATTR(extrapolation_value, Float, 0) | |||
.ATTR(method, String, "bilinear") | |||
.ATTR(dtype, Type, DT_FLOAT) | |||
.OP_END_FACTORY_REG(CropAndResizeV2) | |||
/** | |||
*@brief Extracts crops from the input image tensor and resizes them. | |||
* Extracts crops from the input image tensor and resizes them using bilinear sampling or | |||
* nearest neighbor sampling to a common output size specified by crop_size . \n | |||
@@ -2395,6 +2445,50 @@ REG_OP(EncodeJpegVariableQuality) | |||
.OP_END_FACTORY_REG(EncodeJpegVariableQuality) | |||
/** | |||
* @brief Select top 'pre_nms_topn' scoring boxes, decodes them with respect to anchors, applies non-maximal suppression | |||
* on overlapping boxes with higher than 'nms_threshold' intersection-over-union (IoU) value, discarding boxes where | |||
* shorter side is less than 'min_size'. \n | |||
* @par Inputs: | |||
* scores: 4-D tensor with shape of [num_images, height, width, num_anchors], containing the scores of the boxes for | |||
* given anchors, can be unsorted. Must be one of the following types: float32 . \n | |||
* bbox_deltas: 4-D tensor with shape of [num_images, height, width, 4 * num_anchors], encoding boxes with respect to | |||
* each anchor. Coordinates are given in the form [dy, dx, dh, dw]. Must be one of the following types: float32.. \n | |||
* image_info: 2-D tensor with shape of [num_images, 5], containing image information Height, Width, Scale. Must be one | |||
* of the following types: float32. \n | |||
* anchors: 3-D tensor with shape of [height, width, 4 * num_anchors], describing the anchor boxes. Boxes are formatted | |||
* in the form [y1, x1, y2, x2]. Must be one of the following types: float32. \n | |||
* nms_threshold: A scalar of type float32, non-maximal suppression threshold. \n | |||
* pre_nms_topn: A scalar of type int32, number of top scoring boxes to be used as input. \n | |||
* min_size: A scalar of type float32, Any boxes that has a smaller size than min_size will be discarded. \n | |||
* @par Attributes: | |||
* post_nms_topn: An optional float. Maximum number of rois in the output. Defaults to be 300. \n | |||
* @par Outputs: | |||
* @li rois: 3-D tensor with shape of [num_images, post_nms_topn, 4], padded by 0 if less than 'post_nms_topn'. Must be | |||
* one of the following types: float32 . \n | |||
* @li rois_probabilities: 2-D tensor with shape of [num_images, post_nms_topn], probability of each roi in 'rois' | |||
* padded by 0 if needed, sorted by scores. Must be one of the following types: float32. \n | |||
* @attention Constraints:+ | |||
* Only supports 2864 input boxes at one time.\n | |||
*/ | |||
REG_OP(GenerateBoundingBoxProposals) | |||
.INPUT(scores, TensorType({DT_FLOAT})) | |||
.INPUT(bbox_deltas, TensorType({DT_FLOAT})) | |||
.INPUT(image_info, TensorType({DT_FLOAT})) | |||
.INPUT(anchors, TensorType({DT_FLOAT})) | |||
.INPUT(nms_threshold, TensorType({DT_FLOAT})) | |||
.INPUT(pre_nms_topn, TensorType({DT_INT32})) | |||
.INPUT(min_size, TensorType({DT_FLOAT})) | |||
.OUTPUT(rois, TensorType({DT_FLOAT})) | |||
.OUTPUT(rois_probabilities, TensorType({DT_FLOAT})) | |||
.ATTR(post_nms_topn, Int, 300) | |||
.OP_END_FACTORY_REG(GenerateBoundingBoxProposals) | |||
/** | |||
* @brief image to transforms. \n | |||
* @par Inputs: | |||
@@ -261,11 +261,11 @@ sorted and can be repeated . \n | |||
REG_OP(SparseSegmentSum) | |||
.INPUT(x, TensorType({DT_INT8, DT_UINT8, DT_INT16, DT_UINT16, | |||
DT_INT32, DT_INT64, DT_DOUBLE, DT_FLOAT, DT_FLOAT16})) | |||
.INPUT(indices, TensorType({DT_INT32})) | |||
.INPUT(segment_ids, TensorType({DT_INT32})) | |||
DT_INT32, DT_UINT32, DT_INT64, DT_UINT64, DT_DOUBLE, DT_FLOAT, DT_FLOAT16})) | |||
.INPUT(indices, TensorType({DT_INT32, DT_INT64})) | |||
.INPUT(segment_ids, TensorType({DT_INT32, DT_INT64})) | |||
.OUTPUT(y, TensorType({DT_INT8, DT_UINT8, DT_INT16, DT_UINT16, | |||
DT_INT32, DT_INT64, DT_DOUBLE, DT_FLOAT, DT_FLOAT16})) | |||
DT_INT32, DT_UINT32, DT_INT64, DT_UINT64, DT_DOUBLE, DT_FLOAT, DT_FLOAT16})) | |||
.OP_END_FACTORY_REG(SparseSegmentSum) | |||
/** | |||
@@ -287,10 +287,10 @@ sorted and can be repeated . \n | |||
*/ | |||
REG_OP(SparseSegmentMean) | |||
.INPUT(x, TensorType({DT_FLOAT, DT_DOUBLE})) | |||
.INPUT(indices, TensorType({DT_INT32})) | |||
.INPUT(segment_ids, TensorType({DT_INT32})) | |||
.OUTPUT(y, TensorType({DT_FLOAT, DT_DOUBLE})) | |||
.INPUT(x, TensorType({DT_FLOAT, DT_DOUBLE, DT_FLOAT16})) | |||
.INPUT(indices, TensorType({DT_INT32, DT_INT64})) | |||
.INPUT(segment_ids, TensorType({DT_INT32, DT_INT64})) | |||
.OUTPUT(y, TensorType({DT_FLOAT, DT_DOUBLE, DT_FLOAT16})) | |||
.OP_END_FACTORY_REG(SparseSegmentMean) | |||
/** | |||
@@ -315,11 +315,11 @@ SparseSegmentMean op . \n | |||
*/ | |||
REG_OP(SparseSegmentMeanGrad) | |||
.INPUT(x, TensorType({DT_FLOAT, DT_DOUBLE})) | |||
.INPUT(indices, TensorType({DT_INT32})) | |||
.INPUT(segment_ids, TensorType({DT_INT32})) | |||
.INPUT(x, TensorType({DT_FLOAT, DT_DOUBLE, DT_FLOAT16})) | |||
.INPUT(indices, TensorType({DT_INT32, DT_INT64})) | |||
.INPUT(segment_ids, TensorType({DT_INT32, DT_INT64})) | |||
.INPUT(output_dim0, TensorType({DT_INT32})) | |||
.OUTPUT(y, TensorType({DT_FLOAT, DT_DOUBLE})) | |||
.OUTPUT(y, TensorType({DT_FLOAT, DT_DOUBLE, DT_FLOAT16})) | |||
.OP_END_FACTORY_REG(SparseSegmentMeanGrad) | |||
/** | |||
@@ -211,7 +211,7 @@ REG_OP(SigmoidCrossEntropyWithLogitsV2) | |||
* reduction type to be applied to the output. Defaults to "mean". \n | |||
* @par Outputs: | |||
* loss: Sigmoid focal loss between the predicted value and target value. Has the same dimensions as "pred". \n | |||
* y: Sigmoid focal loss between the predicted value and target value. Has the same dimensions as "pred". \n | |||
* @par Third-party framework compatibility | |||
* Compatible with mmcv operator SigmoidFocalLoss. | |||
@@ -220,13 +220,47 @@ REG_OP(SigmoidFocalLoss) | |||
.INPUT(pred, TensorType({DT_FLOAT16,DT_FLOAT})) | |||
.INPUT(target, TensorType({DT_INT32})) | |||
.OPTIONAL_INPUT(weight, TensorType({DT_FLOAT16, DT_FLOAT})) | |||
.OUTPUT(loss, TensorType({DT_FLOAT16,DT_FLOAT})) | |||
.OUTPUT(y, TensorType({DT_FLOAT16,DT_FLOAT})) | |||
.ATTR(gamma, Float, 2.0) | |||
.ATTR(alpha, Float, 0.25) | |||
.ATTR(reduction, String, "mean") | |||
.OP_END_FACTORY_REG(SigmoidFocalLoss) | |||
/** | |||
* @brief Computes the softmax focal loss of "pred" and "target". | |||
* @par Inputs: | |||
* Three inputs, including: | |||
* @li pred: A 2-dimensional Tensor of type float16 or float32, specifying the predicted value. | |||
* @li target: A 1-dimensional Tensor of type int32, specifying the target value. | |||
* @li weight: A 1-dimensional Tensor, specifying the weight value on class_wise. \n | |||
* @par Attributes: | |||
* @li gamma: An optional float, specifying the exponent of the modulating factor (1 - pt) | |||
* to balance easy/hard examples. Defaults to 2.0. | |||
* @li alpha: An optional float, specifying the weighting factor in range (1, 0) to balance | |||
* the importance of positive/negative examples or less than 0 for ignore. Defaults to 0.25. | |||
* @li reduction: A optional character string from "none", "mean", and "sum", specifying the | |||
* reduction type to be applied to the output. Defaults to "mean". reduction only support | |||
* "none" currently for matching mmcv.\n | |||
* @par Outputs: | |||
* y: Softmax focal loss between the predicted value and target value. Has the same dimensions as "pred". \n | |||
* @par Third-party framework compatibility | |||
* Compatible with mmcv operator SoftmaxFocalLoss. | |||
*/ | |||
REG_OP(SoftmaxFocalLoss) | |||
.INPUT(pred, TensorType({DT_FLOAT16,DT_FLOAT})) | |||
.INPUT(target, TensorType({DT_INT32})) | |||
.OPTIONAL_INPUT(weight, TensorType({DT_FLOAT16, DT_FLOAT})) | |||
.OUTPUT(y, TensorType({DT_FLOAT16,DT_FLOAT})) | |||
.ATTR(gamma, Float, 2.0) | |||
.ATTR(alpha, Float, 0.25) | |||
.ATTR(reduction, String, "mean") | |||
.OP_END_FACTORY_REG(SoftmaxFocalLoss) | |||
/** | |||
* @brief Computes the regression box of the RPN. It is a FasterRCNN operator . | |||
*@par Inputs: | |||
@@ -885,10 +919,10 @@ REG_OP(LNDropoutGrad) | |||
* @li x: A mutable Tensor. Must be one of the following types: | |||
* float16, float32 | |||
* @li mask: A mutable Tensor. Must met all of the following rules: | |||
* shape of mask should be 1D. | |||
* dtype of mask should be uint8. | |||
* value of shape should met the following algorithm: | |||
* value = (size(x) + 128 - 1) // 128 * 128 //8 | |||
* dtype of mask should be uint8 or uint1. | |||
* if data type of mask is uint8, shape of mask should be 1D. value of shape should met the following algorithm: | |||
* value = (size(x) + 128 - 1) // 128 * 128 // 8 | |||
* if data type of mask is uint1, shape of mask should be same to x. | |||
* @li keep_prob: A mutable Tensor. Must met all of the following rules: | |||
* shape of "keep_prob" should be (1,) or [1,]. | |||
* Has the same type as "x" . \n | |||
@@ -898,7 +932,7 @@ REG_OP(LNDropoutGrad) | |||
*/ | |||
REG_OP(DropOutDoMask) | |||
.INPUT(x, TensorType({DT_FLOAT, DT_FLOAT16})) | |||
.INPUT(mask, TensorType({DT_UINT8})) | |||
.INPUT(mask, TensorType({DT_UINT8, DT_UINT1})) | |||
.INPUT(keep_prob, TensorType({DT_FLOAT, DT_FLOAT16})) | |||
.OUTPUT(y, TensorType({DT_FLOAT, DT_FLOAT16})) | |||
.OP_END_FACTORY_REG(DropOutDoMask) | |||
@@ -242,7 +242,7 @@ REG_OP(AvgPool3DD) | |||
.INPUT(x, TensorType({DT_FLOAT16, DT_FLOAT, DT_DOUBLE})) | |||
.OPTIONAL_INPUT(filter, TensorType({DT_FLOAT16, DT_FLOAT, DT_DOUBLE})) | |||
.OPTIONAL_INPUT(multiplier, TensorType({DT_FLOAT16, DT_FLOAT, DT_DOUBLE})) | |||
.OUTPUT(y, TensorType({DT_FLOAT, DT_FLOAT, DT_DOUBLE})) | |||
.OUTPUT(y, TensorType({DT_FLOAT16, DT_FLOAT, DT_DOUBLE})) | |||
.REQUIRED_ATTR(ksize, ListInt) | |||
.REQUIRED_ATTR(strides, ListInt) | |||
.REQUIRED_ATTR(pads, ListInt) | |||
@@ -565,6 +565,7 @@ include: | |||
* @li seed:If seed is set to be -1, and offset is set to be 0, the random number | |||
* generator is seeded by arandom seed. Otherwise, it is seeded by the given seed. | |||
* @li offset:To avoid seed collision . \n | |||
* @li dtype: The type of the output. | |||
* @par Outputs: | |||
* y:A Tensor. A Tensor of type int8, uint8, int16, uint16, | |||
@@ -577,6 +578,7 @@ REG_OP(StatelessBernoulli) | |||
.INPUT(offset, TensorType({ DT_INT64 })) | |||
.OUTPUT(y, TensorType({ DT_INT8, DT_UINT8, DT_INT16, DT_UINT16, DT_INT32, DT_UINT32, | |||
DT_INT64, DT_UINT64, DT_BOOL, DT_FLOAT16, DT_FLOAT, DT_DOUBLE, DT_BF16})) | |||
.ATTR(dtype, Type, DT_FLOAT) | |||
.OP_END_FACTORY_REG(StatelessBernoulli) | |||
/** | |||
*@brief Generates values in an interval . \n | |||
@@ -949,5 +951,35 @@ REG_OP(StatelessRandperm) | |||
.ATTR(dtype, Type, DT_INT64) | |||
.OP_END_FACTORY_REG(StatelessRandperm) | |||
/* | |||
* @brief Generate random bool or uint1 mask for dropout v4 . \n | |||
* @par Inputs: | |||
include: | |||
* @li shape:The shape of the output tensor. | |||
* @li prob:0-D. Prob of 1 . \n | |||
* @par Attributes: | |||
* @li seed:If either seed or seed2 are set to be non-zero, the random number | |||
* generator is seeded by the given seed. Otherwise, it is seeded by a random seed. | |||
* @li seed2:A second seed to avoid seed collision. | |||
* @li dtype:dtype of out put tensor, default is bool . \n | |||
* @par Outputs: | |||
* y:Output random number using bool or uint1 data format . \n | |||
* @par Restrictions: | |||
* Warning: THIS FUNCTION IS EXPERIMENTAL. Please do not use. | |||
* @see DropOutGenMaskV4() | |||
*/ | |||
REG_OP(DropOutGenMaskV4) | |||
.INPUT(shape, TensorType({ DT_INT32, DT_INT64 })) | |||
.INPUT(prob, TensorType({ DT_FLOAT16, DT_FLOAT })) | |||
.OUTPUT(y, TensorType({ DT_BOOL, DT_UINT1 })) | |||
.ATTR(seed, Int, 0) | |||
.ATTR(seed2, Int, 0) | |||
.ATTR(dtype, Type, DT_BOOL) | |||
.OP_END_FACTORY_REG(DropOutGenMaskV4) | |||
} // namespace ge | |||
#endif // OPS_BUILT_IN_OP_PROTO_INC_RANDOM_OPS_H_ |
@@ -261,10 +261,10 @@ REG_OP(GatherV2D) | |||
*/ | |||
REG_OP(GatherElements) | |||
.INPUT(x, TensorType({DT_FLOAT16,DT_FLOAT,DT_INT8,DT_INT16,DT_INT32, | |||
DT_INT64,DT_UINT8,DT_UINT16,DT_UINT32,DT_UINT64})) | |||
DT_INT64,DT_UINT8,DT_UINT16,DT_UINT32,DT_UINT64,DT_BOOL})) | |||
.INPUT(index, TensorType({DT_INT32,DT_INT64})) | |||
.OUTPUT(y, TensorType({DT_FLOAT16,DT_FLOAT,DT_INT8,DT_INT16,DT_INT32, | |||
DT_INT64,DT_UINT8,DT_UINT16,DT_UINT32,DT_UINT64})) | |||
DT_INT64,DT_UINT8,DT_UINT16,DT_UINT32,DT_UINT64,DT_BOOL})) | |||
.ATTR(dim, Int, 0) | |||
.OP_END_FACTORY_REG(GatherElements) | |||
@@ -2366,7 +2366,7 @@ REG_OP(SliceLastDim) | |||
* @par Attributes: | |||
* @li begin_mask: A Tensor of type int32. | |||
* Developers can ignore this attribute. | |||
* A bitmask where a bit "i" being "1" means to ignore the begin | |||
* A bitmask where a bit "i" being "1" means to ignore the begin | |||
* value and instead use the largest interval possible. | |||
* @li end_mask: A Tensor of type int32. | |||
* Developers can ignore this attribute. | |||
@@ -2377,7 +2377,7 @@ REG_OP(SliceLastDim) | |||
* is actually an ellipsis. | |||
* @li new_axis_mask: A Tensor of type int32. | |||
* Developers can ignore this attribute. | |||
* A bitmask where bit "i" being "1" means the "i"th | |||
* A bitmask where bit "i" being "1" means the "i"th | |||
* specification creates a new shape 1 dimension. | |||
* @li shrink_axis_mask: A Tensor of type int32. | |||
* Developers can ignore this attribute. | |||
@@ -51,10 +51,10 @@ typedef enum tagRtDeviceCapabilityType { | |||
RT_SCHEDULE_SOFTWARE_OPT, | |||
RT_SCHEDULE_HARDWARE, // HWTS Schedule | |||
RT_AICPU_BLOCKING_OP_NOT_SUPPORT, | |||
RT_AICPU_BLOCKING_OP_SUPPORT, // 1910/1980/1951 ts support AICPU blocking operation | |||
RT_AICPU_BLOCKING_OP_SUPPORT, // 1910/1980/51 ts support AICPU blocking operation | |||
RT_MODE_NO_FFTS, // no ffts | |||
RT_MODE_FFTS, // 1981 get ffts work mode, ffts | |||
RT_MODE_FFTS_PLUS, // 1981 get ffts work mode, ffts plus | |||
RT_MODE_FFTS, // 81 get ffts work mode, ffts | |||
RT_MODE_FFTS_PLUS, // 81 get ffts work mode, ffts plus | |||
} rtDeviceCapabilityType; | |||
typedef enum tagRtVersion { | |||
@@ -82,9 +82,11 @@ typedef enum tagRtPlatformType { | |||
PLATFORM_CLOUD_V2_910B1 = 10, | |||
PLATFORM_CLOUD_V2_910B2 = 11, | |||
PLATFORM_CLOUD_V2_910B3 = 12, | |||
PLATFORM_CLOUD_V2_910B4 = 13, | |||
PLATFORM_MDC_PG2 = 14, | |||
PLATFORM_END = 15, | |||
PLATFORM_CLOUD_V2_910C1 = 13, | |||
PLATFORM_CLOUD_V2_910C2 = 14, | |||
PLATFORM_CLOUD_V2_910C3 = 15, | |||
PLATFORM_MDC_BS9SX1A = 16, | |||
PLATFORM_END = 17, | |||
} rtPlatformType_t; | |||
typedef enum tagRtCubeFracMKNFp16 { | |||
@@ -79,9 +79,14 @@ typedef enum tagRtDeviceModuleType { | |||
RT_MODULE_TYPE_AICORE, /**< AI CORE info*/ | |||
RT_MODULE_TYPE_TSCPU, /**< tscpu info*/ | |||
RT_MODULE_TYPE_PCIE, /**< PCIE info*/ | |||
RT_MODULE_TYPE_VECTOR_CORE, /**< VECTOR CORE info*/ | |||
RT_MODULE_TYPE_VECTOR_CORE /**< VECTOR CORE info*/ | |||
} rtDeviceModuleType_t; | |||
typedef enum tagRtPhyDeviceInfoType { | |||
RT_PHY_INFO_TYPE_CHIPTYPE = 0, | |||
RT_PHY_INFO_TYPE_MASTER_ID | |||
} rtPhyDeviceInfoType_t; | |||
typedef enum tagRtMemRequestFeature { | |||
MEM_REQUEST_FEATURE_DEFAULT = 0, | |||
MEM_REQUEST_FEATURE_OPP, | |||
@@ -149,6 +154,18 @@ RTS_API rtError_t rtGetDeviceIDs(uint32_t *devices, uint32_t len); | |||
RTS_API rtError_t rtGetDeviceInfo(uint32_t deviceId, int32_t moduleType, int32_t infoType, int64_t *val); | |||
/** | |||
* @ingroup dvrt_dev | |||
* @brief get phy device infomation. | |||
* @param [int] phyId the physic Id | |||
* @param [int] moduleType module type | |||
* @param [int] infoType info type | |||
* @param [out] val the device info | |||
* @return RT_ERROR_NONE for ok | |||
* @return RT_ERROR_DRV_ERR for error | |||
*/ | |||
RTS_API rtError_t rtGetPhyDeviceInfo(uint32_t phyId, int32_t moduleType, int32_t infoType, int64_t *val); | |||
/** | |||
* @ingroup dvrt_dev | |||
* @brief set target device for current thread | |||
* @param [int] devId the device id | |||
@@ -19,7 +19,7 @@ extern "C" { | |||
* @brief shared memory data control | |||
*/ | |||
typedef struct tagRtSmData { | |||
uint64_t L2_mirror_addr; // preload or swap source address | |||
uint64_t L2_mirror_addr; // preload or swap source addr | |||
uint32_t L2_data_section_size; // every data size | |||
uint8_t L2_preload; // 1 - preload from mirrorAddr, 0 - no preload | |||
uint8_t modified; // 1 - data will be modified by kernel, 0 - no modified | |||
@@ -377,7 +377,7 @@ RTS_API rtError_t rtQueryFunctionRegistered(const char_t *stubName); | |||
* @brief config data dump | |||
* @param [in] dumpSizePerBlock dump size | |||
* @param [in] blockDim block dimentions | |||
* @param [in] dumpBaseAddr dump base address | |||
* @param [in] dumpBaseAddr dump base addr | |||
* @return RT_ERROR_NONE for ok | |||
* @return RT_ERROR_INVALID_VALUE for error input | |||
*/ | |||
@@ -577,15 +577,26 @@ RTS_API rtError_t rtDatadumpInfoLoad(const void *dumpInfo, uint32_t length); | |||
/** | |||
* @ingroup rt_kernel | |||
* @brief load dump info to aicpu | |||
* @param [in] dumpInfo dump info | |||
* @param [in] length length of dump info | |||
* @param [in] flag RT_KERNEL_DEFAULT or RT_KERNEL_CUSTOM_AICPU | |||
* @return RT_ERROR_NONE for ok | |||
* @return RT_ERROR_INVALID_VALUE for error input | |||
*/ | |||
RTS_API rtError_t rtDatadumpInfoLoadWithFlag(const void *dumpInfo, const uint32_t length, const uint32_t flag); | |||
/** | |||
* @ingroup rt_kernel | |||
* @brief launch npu get float status task | |||
* @param [in] outputAddr pointer to op output addr | |||
* @param [in] outputAddrPtr pointer to op output addr | |||
* @param [in] outputSize op output size | |||
* @param [in] checkMode check mode | |||
* @param [in] stm associated stream | |||
* @return RT_ERROR_NONE for ok | |||
* @return RT_ERROR_INVALID_VALUE for error input | |||
*/ | |||
RTS_API rtError_t rtNpuGetFloatStatus(void *outputAddr, uint64_t outputSize, uint32_t checkMode, rtStream_t stm); | |||
RTS_API rtError_t rtNpuGetFloatStatus(void *outputAddrPtr, uint64_t outputSize, uint32_t checkMode, rtStream_t stm); | |||
/** | |||
* @ingroup rt_kernel | |||
@@ -29,7 +29,7 @@ extern "C" { | |||
#define RT_MEMORY_P2P_DDR (0x11U) // DDR memory on other device | |||
#define RT_MEMORY_DDR_NC (0x20U) // DDR memory of non-cache | |||
#define RT_MEMORY_TS (0x40U) // Used for Ts memory | |||
#define RT_MEMORY_TS_4G (0x40U) // Used for Ts memory(only 1951) | |||
#define RT_MEMORY_TS_4G (0x40U) // Used for Ts memory(only 51) | |||
#define RT_MEMORY_HOST (0x81U) // Memory on host | |||
#define RT_MEMORY_SVM (0x90U) // Memory for SVM | |||
#define RT_MEMORY_HOST_SVM (0x90U) // Memory for host SVM | |||
@@ -222,6 +222,7 @@ typedef struct { | |||
*/ | |||
RTS_API rtError_t rtMalloc(void **devPtr, uint64_t size, rtMemType_t type); | |||
RTS_API rtError_t rtMallocV2(void **devPtr, uint64_t size, rtMemType_t type, const uint16_t moduleId); | |||
/** | |||
* @ingroup dvrt_mem | |||
* @brief free device memory | |||
@@ -241,6 +242,8 @@ RTS_API rtError_t rtFree(void *devPtr); | |||
*/ | |||
RTS_API rtError_t rtDvppMalloc(void **devPtr, uint64_t size); | |||
rtError_t rtDvppMallocV2(void **devPtr, uint64_t size, uint16_t moduleId); | |||
/** | |||
* @ingroup dvrt_mem | |||
* @brief alloc device memory for dvpp, support set flag | |||
@@ -253,6 +256,8 @@ RTS_API rtError_t rtDvppMalloc(void **devPtr, uint64_t size); | |||
*/ | |||
RTS_API rtError_t rtDvppMallocWithFlag(void **devPtr, uint64_t size, uint32_t flag); | |||
RTS_API rtError_t rtDvppMallocWithFlagV2(void **devPtr, uint64_t size, uint32_t flag, const uint16_t moduleId); | |||
/** | |||
* @ingroup dvrt_mem | |||
* @brief free device memory for dvpp | |||
@@ -272,6 +277,8 @@ RTS_API rtError_t rtDvppFree(void *devPtr); | |||
*/ | |||
RTS_API rtError_t rtMallocHost(void **hostPtr, uint64_t size); | |||
RTS_API rtError_t rtMallocHostV2(void **hostPtr, uint64_t size, uint16_t moduleId); | |||
/** | |||
* @ingroup dvrt_mem | |||
* @brief free host memory | |||
@@ -314,6 +321,8 @@ RTS_API rtError_t rtFreeHostSharedMemory(rtFreeHostSharedMemoryIn *in); | |||
*/ | |||
RTS_API rtError_t rtMemAllocManaged(void **ptr, uint64_t size, uint32_t flag); | |||
RTS_API rtError_t rtMemAllocManagedV2(void **ptr, uint64_t size, uint32_t flag, uint16_t moduleId); | |||
/** | |||
* @ingroup dvrt_mem | |||
* @brief free managed memory | |||
@@ -333,10 +342,12 @@ RTS_API rtError_t rtMemFreeManaged(void *ptr); | |||
*/ | |||
RTS_API rtError_t rtMallocCached(void **devPtr, uint64_t size, rtMemType_t type); | |||
RTS_API rtError_t rtMallocCachedV2(void **devPtr, uint64_t size, rtMemType_t type, uint16_t moduleId); | |||
/** | |||
* @ingroup dvrt_mem | |||
* @brief flush device mempory | |||
* @param [in] base virtal base address | |||
* @param [in] base virtal base addr | |||
* @param [in] len memory size | |||
* @return RT_ERROR_NONE for ok, errno for failed | |||
*/ | |||
@@ -345,7 +356,7 @@ RTS_API rtError_t rtFlushCache(void *base, size_t len); | |||
/** | |||
* @ingroup dvrt_mem | |||
* @brief invalid device mempory | |||
* @param [in] base virtal base address | |||
* @param [in] base virtal base addr | |||
* @param [in] len memory size | |||
* @return RT_ERROR_NONE for ok, errno for failed | |||
*/ | |||
@@ -399,15 +410,15 @@ RTS_API rtError_t rtMemcpyAsync(void *dst, uint64_t destMax, const void *src, ui | |||
* @param [in] dst destination address pointer | |||
* @param [in] Max length of destination address memory | |||
* @param [in] src source address pointer | |||
* @param [in] count the number of byte to copy | |||
* @param [in] cnt the number of byte to copy | |||
* @param [in] kind memcpy type | |||
* @param [in] stream asynchronized task stream | |||
* @param [in] stm asynchronized task stream | |||
* @param [in] qosCfg asynchronized task qosCfg | |||
* @return RT_ERROR_NONE for ok | |||
* @return RT_ERROR_INVALID_VALUE for error input | |||
*/ | |||
RTS_API rtError_t rtMemcpyAsyncWithCfg(void *dst, uint64_t destMax, const void *src, uint64_t count, | |||
rtMemcpyKind_t kind, rtStream_t stream, uint32_t qosCfg); | |||
RTS_API rtError_t rtMemcpyAsyncWithCfg(void *dst, uint64_t destMax, const void *src, uint64_t cnt, | |||
rtMemcpyKind_t kind, rtStream_t stm, uint32_t qosCfg); | |||
typedef struct { | |||
uint32_t resv0; | |||
@@ -402,6 +402,16 @@ RTS_API rtError_t rtMbufSetDataLen(rtMbufPtr_t memBuf, uint64_t len); | |||
/** | |||
* @ingroup rt_mem_queue | |||
* @brief set Data len of Mbuf | |||
* @param [in] memBuf: Mbuf addr | |||
* @param [out] len: data len | |||
* @return RT_ERROR_NONE for success, others for fail | |||
*/ | |||
RTS_API rtError_t rtMbufGetDataLen(rtMbufPtr_t memBuf, uint64_t *len); | |||
/** | |||
* @ingroup rt_mem_queue | |||
* @brief get Data addr of Mbuf | |||
* @param [in] memBuf: Mbuf addr | |||
* @param [out] buf: Mbuf data addr | |||
@@ -437,6 +447,34 @@ RTS_API rtError_t rtMbufGetPrivInfo(rtMbufPtr_t memBuf, void **priv, uint64_t * | |||
*/ | |||
RTS_API rtError_t rtMbufCopyBufRef(rtMbufPtr_t memBuf, rtMbufPtr_t *newMemBuf); | |||
/** | |||
* @ingroup rt_mem_queue | |||
* @brief append mbuf to mbuf chain | |||
* @param [inout] memBufChainHead, the mbuf chain head | |||
* @param [in] memBuf, the mbuf to append | |||
* @return RT_ERROR_NONE for ok | |||
*/ | |||
RTS_API rtError_t rtMbufChainAppend(rtMbufPtr_t memBufChainHead, rtMbufPtr_t memBuf); | |||
/** | |||
* @ingroup rt_mem_queue | |||
* @brief get mbuf num in mbuf chain | |||
* @param [in] memBufChainHead, the mbuf chain head | |||
* @param [out] num, the mbuf chain size | |||
* @return RT_ERROR_NONE for ok | |||
*/ | |||
RTS_API rtError_t rtMbufChainGetMbufNum(rtMbufPtr_t memBufChainHead, uint32_t *num); | |||
/** | |||
* @ingroup rt_mem_queue | |||
* @brief get mbuf in mbuf chain | |||
* @param [in] mbufChainHead, the mbuf chain head | |||
* @param [in] index, the mbuf index which to get in chain | |||
* @param [out] mbuf, the mbuf to get | |||
* @return RT_ERROR_NONE for ok | |||
*/ | |||
RTS_API rtError_t rtMbufChainGetMbuf(rtMbufPtr_t memBufChainHead, uint32_t index, rtMbufPtr_t *memBuf); | |||
// mem group | |||
typedef struct { | |||
uint64_t maxMemSize; // max buf size in grp, in KB. = 0 means no limit | |||
@@ -23,6 +23,17 @@ extern "C" { | |||
*/ | |||
RTS_API rtError_t rtStarsTaskLaunch(const void *taskSqe, uint32_t sqeLen, rtStream_t stm); | |||
/** | |||
* @ingroup rt_stars | |||
* @brief launch stars task. | |||
* used for send star sqe directly. | |||
* @param [in] taskSqe stars task sqe | |||
* @param [in] sqeLen stars task sqe length | |||
* @param [in] stm associated stream | |||
* @param [in] flag dump flag | |||
* @return RT_ERROR_NONE for ok, others failed | |||
*/ | |||
RTS_API rtError_t rtStarsTaskLaunchWithFlag(const void *taskSqe, uint32_t sqeLen, rtStream_t stm, uint32_t flag); | |||
/** | |||
* @ingroup rt_stars | |||
@@ -28,6 +28,7 @@ extern "C" { | |||
#define RT_STREAM_PRIMARY_DEFAULT (0x40U) | |||
#define RT_STREAM_PRIMARY_FIRST_DEFAULT (0x80U) | |||
#define RT_STREAM_OVERFLOW (0x100U) | |||
#define RT_STREAM_FAST_LAUNCH (0x200U) | |||
/** | |||
* @ingroup stream_type | |||
@@ -73,6 +74,15 @@ RTS_API rtError_t rtStreamDestroy(rtStream_t stm); | |||
/** | |||
* @ingroup dvrt_stream | |||
* @brief force destroy stream instance. | |||
* @param [in] stm the stream to destroy | |||
* @return RT_ERROR_NONE for ok | |||
* @return RT_ERROR_INVALID_VALUE for error input | |||
*/ | |||
RTS_API rtError_t rtStreamDestroyForce(rtStream_t stm); | |||
/** | |||
* @ingroup dvrt_stream | |||
* @brief wait an recorded event for stream | |||
* @param [in] stm the wait stream | |||
* @param [in] event the event to wait | |||
@@ -83,10 +93,10 @@ RTS_API rtError_t rtStreamWaitEvent(rtStream_t stm, rtEvent_t evt); | |||
/** | |||
* @ingroup dvrt_stream | |||
* @brief wait an recorded event for stream, used for 1951 pg1 | |||
* @brief wait an recorded event for stream, used for 51 pg1 | |||
* @param [in] stm the wait stream | |||
* @param [in] event the event to wait | |||
* @param [in] timeout timeout value for 1951 pg1 | |||
* @param [in] timeout timeout value for 51 pg1 | |||
* @return RT_ERROR_NONE for ok | |||
* @return RT_ERROR_INVALID_VALUE for error input | |||
*/ | |||
@@ -209,6 +209,7 @@ enum { | |||
OP, | |||
UDF, | |||
HICAID, | |||
TSYNC, | |||
INVLID_MOUDLE_ID | |||
}; | |||