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

Update ops-related pbtxt files.

Browse Source 
PiperOrigin-RevId: 358292541
Change-Id: Ieb03481abb80ee0e0c0e295fcf4e0726d27f4e02
This commit is contained in:
A. Unique TensorFlower 2021-02-18 16:36:49 -08:00 committed by TensorFlower Gardener
parent be625d2fe8
commit 3b4f970c3b
2 changed files with 44 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
tensorflow/core/ops/compat/ops_history_v2/TPUReshardVariables.pbtxt Normal file
View File

@ -0,0 +1,22 @@
op {
name: "TPUReshardVariables"
input_arg {
name: "vars"
type: DT_RESOURCE
number_attr: "N"
}
input_arg {
name: "new_format_key"
type: DT_STRING
}
input_arg {
name: "format_state_var"
type: DT_RESOURCE
}
attr {
name: "N"
type: "int"
has_minimum: true
}
is_stateful: true
}

22
tensorflow/core/ops/ops.pbtxt
View File

@ -54108,6 +54108,28 @@ op {
type: "type"
}
}
op {
name: "TPUReshardVariables"
input_arg {
name: "vars"
type: DT_RESOURCE
number_attr: "N"
}
input_arg {
name: "new_format_key"
type: DT_STRING
}
input_arg {
name: "format_state_var"
type: DT_RESOURCE
}
attr {
name: "N"
type: "int"
has_minimum: true
}
is_stateful: true
}
op {
name: "TakeDataset"
input_arg {