Home
last modified time | relevance | path

Searched refs:TPUExecuteAndUpdateVariables (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
H A Dtpu_colocate_composite_resource_ops.mlir22 …"tf.TPUExecuteAndUpdateVariables"(%arg1, %1) {device_var_reads_indices = [0], device_var_updates_i…
53 …"tf.TPUExecuteAndUpdateVariables"(%arg1, %2) {device_var_reads_indices = [0], device_var_updates_i…
87 …"tf.TPUExecuteAndUpdateVariables"(%arg1, %2) {device_var_reads_indices = [0], device_var_updates_i…
112 …"tf.TPUExecuteAndUpdateVariables"(%arg1, %2) {device_var_reads_indices = [0], device_var_updates_i…
H A Dtpu-variable-runtime-reformatting.mlir67 // CHECK: "tf.TPUExecuteAndUpdateVariables"(%[[ID]], %[[R1]], %[[COMPILE]]#1)
69 "tf.TPUExecuteAndUpdateVariables"(%id, %arg31, %compile#1)
146 "tf.TPUExecuteAndUpdateVariables"(%arg30, %arg31, %arg32, %compile#1)
206 "tf.TPUExecuteAndUpdateVariables"(%id, %arg31, %compile#1)
287 // CHECK: "tf.TPUExecuteAndUpdateVariables"(%[[ID]], %[[R1]], %[[COMPILE]]#1)
289 "tf.TPUExecuteAndUpdateVariables"(%id, %arg31, %compile#1)
H A Dtpu-merge-variables-with-execute.mlir30 …// CHECK-NEXT: "tf.TPUExecuteAndUpdateVariables"(%[[ID_0]], %[[ARG_1]], %[[READ_2]], %[[COMPILE]]#…
74 // CHECK-NEXT: "tf.TPUExecuteAndUpdateVariables"(%[[READ_0]], %[[R_ARG]], %[[COMPILE]]#1)
133 …// CHECK-NEXT: "tf.TPUExecuteAndUpdateVariables"(%[[READ_0]], %[[ARG_1]], %[[ARG_4]], %[[READ_5]],…
200 …// CHECK-NEXT: "tf.TPUExecuteAndUpdateVariables"(%[[ARG_0]], %[[ARG_1]], %[[ARG_3]], %[[ARG_4]], %…
305 // CHECK-NEXT: "tf.TPUExecuteAndUpdateVariables"(%[[ARG_0]], %[[ARG_2]])
315 // CHECK-NEXT: "tf.TPUExecuteAndUpdateVariables"(%[[ARG_1]], %[[ARG_2]])
358 // CHECK-NEXT: "tf.TPUExecuteAndUpdateVariables"(%[[RI_0]], %[[ARG_4]])
367 // CHECK-NEXT: "tf.TPUExecuteAndUpdateVariables"(%[[RI_1]], %[[ARG_4]])
H A Dtf-ops.mlir4257 …"tf.TPUExecuteAndUpdateVariables"(%arg0, %arg1) {device_var_reads_indices = [0, 1], device_var_upd…
4265 …"tf.TPUExecuteAndUpdateVariables"(%arg0, %arg1) {device_var_reads_indices = [0], device_var_update…
4273 …"tf.TPUExecuteAndUpdateVariables"(%arg0, %arg1) {device_var_reads_indices = [-1], device_var_updat…
4281 …"tf.TPUExecuteAndUpdateVariables"(%arg0, %arg1) {device_var_reads_indices = [0], device_var_update…
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
H A DTPUExecuteAndUpdateVariables.pbtxt2 name: "TPUExecuteAndUpdateVariables"
38 name: "TPUExecuteAndUpdateVariables"
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/
H A Dapi_def_TPUExecuteAndUpdateVariables.pbtxt2 graph_op_name: "TPUExecuteAndUpdateVariables"
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtf_passes.td1594 `tf.TPUExecute` op and merges them into a `tf.TPUExecuteAndUpdateVariables`
1610 %2 = "tf.TPUExecuteAndUpdateVariables"(%arg0, %arg1, %compile)
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
H A Dlegalization.md3119 ### tf.TPUExecuteAndUpdateVariables
/aosp_15_r20/external/tensorflow/tensorflow/go/op/
H A Dwrappers.go50189 func TPUExecuteAndUpdateVariables(scope *Scope, args []tf.Output, key tf.Output, Tresults []tf.Data… func
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dops.pbtxt57603 name: "TPUExecuteAndUpdateVariables"
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_generated_ops.td17662 def TF_TPUExecuteAndUpdateVariablesOp : TF_Op<"TPUExecuteAndUpdateVariables", [DeclareOpInterfaceMe…