Add auto-generated LookupTableFindV2, LookupTableImportV2, and LookupTableSizeV2 ops.
PiperOrigin-RevId: 281679937 Change-Id: I28d3bd53cc3392cb83768ed3838484b0f2860e1d
This commit is contained in:
parent
c5dce1fd00
commit
0c2b12c072
@ -2441,6 +2441,68 @@ def TF_LogicalOrOp : TF_Op<"LogicalOr", [Broadcastable, Commutative, NoSideEffec
|
||||
);
|
||||
}
|
||||
|
||||
def TF_LookupTableFindV2Op : TF_Op<"LookupTableFindV2", []> {
|
||||
let summary = "Looks up keys in a table, outputs the corresponding values.";
|
||||
|
||||
let description = [{
|
||||
The tensor `keys` must of the same type as the keys of the table.
|
||||
The output `values` is of the type of the table values.
|
||||
|
||||
The scalar `default_value` is the value output for keys not present in the
|
||||
table. It must also be of the same type as the table values.
|
||||
}];
|
||||
|
||||
let arguments = (ins
|
||||
TF_ResourceTensor:$table_handle,
|
||||
TF_Tensor:$keys,
|
||||
TF_Tensor:$default_value
|
||||
);
|
||||
|
||||
let results = (outs
|
||||
TF_Tensor:$values
|
||||
);
|
||||
|
||||
TF_DerivedOperandTypeAttr Tin = TF_DerivedOperandTypeAttr<1>;
|
||||
TF_DerivedOperandTypeAttr Tout = TF_DerivedOperandTypeAttr<2>;
|
||||
}
|
||||
|
||||
def TF_LookupTableImportV2Op : TF_Op<"LookupTableImportV2", []> {
|
||||
let summary = [{
|
||||
Replaces the contents of the table with the specified keys and values.
|
||||
}];
|
||||
|
||||
let description = [{
|
||||
The tensor `keys` must be of the same type as the keys of the table.
|
||||
The tensor `values` must be of the type of the table values.
|
||||
}];
|
||||
|
||||
let arguments = (ins
|
||||
TF_ResourceTensor:$table_handle,
|
||||
TF_Tensor:$keys,
|
||||
TF_Tensor:$values
|
||||
);
|
||||
|
||||
let results = (outs);
|
||||
|
||||
TF_DerivedOperandTypeAttr Tin = TF_DerivedOperandTypeAttr<1>;
|
||||
TF_DerivedOperandTypeAttr Tout = TF_DerivedOperandTypeAttr<2>;
|
||||
}
|
||||
|
||||
def TF_LookupTableSizeV2Op : TF_Op<"LookupTableSizeV2", []> {
|
||||
let summary = "Computes the number of elements in the given table.";
|
||||
|
||||
let description = [{
|
||||
}];
|
||||
|
||||
let arguments = (ins
|
||||
TF_ResourceTensor:$table_handle
|
||||
);
|
||||
|
||||
let results = (outs
|
||||
I64Tensor:$size
|
||||
);
|
||||
}
|
||||
|
||||
def TF_MatMulOp : TF_Op<"MatMul", [NoSideEffect]> {
|
||||
let summary = [{
|
||||
Multiply the matrix "a" by the matrix "b".
|
||||
|
Loading…
x
Reference in New Issue
Block a user