Browse Source

Adapt to TBE Cast operator latest interface

tags/v0.6.0-beta
梁成辉 6 years ago
parent
commit
22866fbe25
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/transform/op_declare.cc

+ 1
- 1
mindspore/ccsrc/transform/op_declare.cc View File

@@ -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<GEType>())}};
ATTR_MAP(Cast) = {{"Truncate", ATTR_DESC(truncate, AnyTraits<bool>())}};
ATTR_MAP(Cast) = EMPTY_ATTR_MAP;
OUTPUT_MAP(Cast) = {{0, OUTPUT_DESC(y)}};

// Reciprocal


Loading…
Cancel
Save