experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[TF:XLA] Allow fallback to TF:Classic when resource variables have changed shape or type.

Browse Source 
PiperOrigin-RevId: 278868665
Change-Id: Iaec795c460cf7c34160ad503cd9e704fd7453fb4
This commit is contained in:
Tres Popp 2019-11-06 08:46:12 -08:00 committed by TensorFlower Gardener
parent b9be248fd3
commit 965f4b71f0
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/compiler/tf2xla/xla_resource.cc
View File

@ -92,14 +92,14 @@ Status XlaResource::SetTypeAndShape(DataType type, const TensorShape& shape) {
"'' to an invalid type"); "'' to an invalid type");
} }
if (initialized() && type_ != type) { if (initialized() && type_ != type) {
return errors::InvalidArgument("Type of resource ", name_, return errors::Unimplemented("Type of resource ", name_,
" cannot be changed after initialization: " " cannot be changed after initialization: "
"old type was ", "old type was ",
DataTypeString(type_), ", new type is ", DataTypeString(type_), ", new type is ",
DataTypeString(type)); DataTypeString(type));
} }
if (initialized() && shape_ != shape) { if (initialized() && shape_ != shape) {
return errors::InvalidArgument("Shape of resource ", name_, return errors::Unimplemented("Shape of resource ", name_,
" cannot be changed after initialization: " " cannot be changed after initialization: "
"old shape was ", "old shape was ",
shape_.DebugString(), ", new shape is ", shape_.DebugString(), ", new shape is ",