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

Populate side-effect decorators for HashTable ops.

Browse Source 
PiperOrigin-RevId: 317044477
Change-Id: I4837078079924cd08d4a4e7e63066f8268d92112
This commit is contained in:
Smit Hinsu 2020-06-17 23:49:02 -07:00 committed by TensorFlower Gardener
parent c2d9ded23c
commit 0a8019fa2b
3 changed files with 3 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
View File

@ -3884,7 +3884,7 @@ table will be immutable.
);
let results = (outs
Res<TF_ResourceTensor, "", [TF_TableRead]>:$table_handle
TF_ResourceTensor:$table_handle
);
}
@ -4836,7 +4836,7 @@ table. It must also be of the same type as the table values.
}];
let arguments = (ins
Arg<TF_ResourceTensor, "", [TF_TableRead]>:$table_handle,
TF_ResourceTensor:$table_handle,
TF_Tensor:$keys,
TF_Tensor:$default_value
);
@ -4860,7 +4860,7 @@ The tensor `values` must be of the type of the table values.
}];
let arguments = (ins
Arg<TF_ResourceTensor, "", [TF_TableWrite]>:$table_handle,
TF_ResourceTensor:$table_handle,
TF_Tensor:$keys,
TF_Tensor:$values
);

3
tensorflow/compiler/mlir/tensorflow/ir/tf_op_base.td
View File

@ -91,17 +91,14 @@ class TF_ResourceBase<string resourceKind> :
def TF_VariableResource : TF_ResourceBase<"Variable">;
def TF_StackResource : TF_ResourceBase<"Stack">;
def TF_TensorArrayResource : TF_ResourceBase<"TensorArray">;
def TF_TableResource : TF_ResourceBase<"Table">;
def TF_VariableRead : MemRead<TF_VariableResource>;
def TF_StackRead : MemRead<TF_StackResource>;
def TF_TensorArrayRead : MemRead<TF_TensorArrayResource>;
def TF_TableRead : MemRead<TF_TableResource>;
def TF_VariableWrite : MemWrite<TF_VariableResource>;
def TF_StackWrite : MemWrite<TF_StackResource>;
def TF_TensorArrayWrite : MemWrite<TF_TensorArrayResource>;
def TF_TableWrite: MemWrite<TF_TableResource>;
//===----------------------------------------------------------------------===//
// TensorFlow op definitions

4
tensorflow/compiler/mlir/tensorflow/ir/tf_side_effects.h
View File

@ -35,10 +35,6 @@ struct TensorArray : ::mlir::SideEffects::Resource::Base<TensorArray> {
StringRef getName() final { return "TensorArray"; }
};
struct Table : ::mlir::SideEffects::Resource::Base<Table> {
StringRef getName() final { return "Table"; }
};
} // namespace ResourceEffects
} // namespace TF
} // namespace mlir