From 22866fbe2577bbb1bcfe2befe15a8acdcb8f3c7a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=A2=81=E6=88=90=E8=BE=89?= Date: Sat, 16 May 2020 12:17:06 +0800 Subject: [PATCH] Adapt to TBE Cast operator latest interface --- mindspore/ccsrc/transform/op_declare.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/ccsrc/transform/op_declare.cc b/mindspore/ccsrc/transform/op_declare.cc index 27c1d306aa..5cae6c77f7 100644 --- a/mindspore/ccsrc/transform/op_declare.cc +++ b/mindspore/ccsrc/transform/op_declare.cc @@ -823,7 +823,7 @@ OUTPUT_MAP(RealDiv) = {{0, OUTPUT_DESC(y)}}; // Cast INPUT_MAP(Cast) = {{1, INPUT_DESC(x)}}; INPUT_ATTR_MAP(Cast) = {{2, ATTR_DESC(dst_type, AnyTraits())}}; -ATTR_MAP(Cast) = {{"Truncate", ATTR_DESC(truncate, AnyTraits())}}; +ATTR_MAP(Cast) = EMPTY_ATTR_MAP; OUTPUT_MAP(Cast) = {{0, OUTPUT_DESC(y)}}; // Reciprocal