fix export print AIR model files

Signed-off-by: zjun <zhangjun0@huawei.com>

fix graphengine commit

Signed-off-by: zjun <zhangjun0@huawei.com>
pull/6158/head
zjun 4 years ago
parent 12f3665167
commit 5bc855416c

@ -17,13 +17,13 @@
#include "transform/graph_ir/op_declare/logging_ops_declare.h" #include "transform/graph_ir/op_declare/logging_ops_declare.h"
namespace mindspore::transform { namespace mindspore::transform {
#ifdef ENABLE_GE
// Print // Print
INPUT_MAP(Print) = EMPTY_INPUT_MAP; INPUT_MAP(Print) = EMPTY_INPUT_MAP;
DYN_INPUT_MAP(Print) = {{1, DYN_INPUT_DESC(x)}}; DYN_INPUT_MAP(Print) = {{1, DYN_INPUT_DESC(x)}};
ATTR_MAP(Print) = EMPTY_ATTR_MAP; ATTR_MAP(Print) = EMPTY_ATTR_MAP;
REG_ADPT_DESC(Print, kNamePrint, ADPT_DESC(Print)) REG_ADPT_DESC(Print, kNamePrint, ADPT_DESC(Print))
#ifdef ENABLE_GE
INPUT_MAP(Assert) = {{1, INPUT_DESC(input_condition)}}; INPUT_MAP(Assert) = {{1, INPUT_DESC(input_condition)}};
DYN_INPUT_MAP(Assert) = {{2, DYN_INPUT_DESC(input_data)}}; DYN_INPUT_MAP(Assert) = {{2, DYN_INPUT_DESC(input_data)}};
ATTR_MAP(Assert) = {{"summarize", ATTR_DESC(summarize, AnyTraits<int>())}}; ATTR_MAP(Assert) = {{"summarize", ATTR_DESC(summarize, AnyTraits<int>())}};

@ -23,10 +23,10 @@
#include "ops/logging_ops.h" #include "ops/logging_ops.h"
namespace mindspore::transform { namespace mindspore::transform {
#ifdef ENABLE_GE
DECLARE_OP_ADAPTER(Print) DECLARE_OP_ADAPTER(Print)
DECLARE_OP_USE_DYN_INPUT(Print) DECLARE_OP_USE_DYN_INPUT(Print)
#ifdef ENABLE_GE
DECLARE_OP_ADAPTER(Assert) DECLARE_OP_ADAPTER(Assert)
DECLARE_OP_USE_DYN_INPUT(Assert) DECLARE_OP_USE_DYN_INPUT(Assert)
#endif #endif

Loading…
Cancel
Save