Searched refs:TPUExecuteAndUpdateVariables (Results 1 – 11 of 11) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
H A D | tpu_colocate_composite_resource_ops.mlir | 22 …"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 D | tpu-variable-runtime-reformatting.mlir | 67 // 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 D | tpu-merge-variables-with-execute.mlir | 30 …// 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 D | tf-ops.mlir | 4257 …"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 D | TPUExecuteAndUpdateVariables.pbtxt | 2 name: "TPUExecuteAndUpdateVariables" 38 name: "TPUExecuteAndUpdateVariables"
|
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/ |
H A D | api_def_TPUExecuteAndUpdateVariables.pbtxt | 2 graph_op_name: "TPUExecuteAndUpdateVariables"
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
H A D | tf_passes.td | 1594 `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 D | legalization.md | 3119 ### tf.TPUExecuteAndUpdateVariables
|
/aosp_15_r20/external/tensorflow/tensorflow/go/op/ |
H A D | wrappers.go | 50189 func TPUExecuteAndUpdateVariables(scope *Scope, args []tf.Output, key tf.Output, Tresults []tf.Data… func
|
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
H A D | ops.pbtxt | 57603 name: "TPUExecuteAndUpdateVariables"
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
H A D | tf_generated_ops.td | 17662 def TF_TPUExecuteAndUpdateVariablesOp : TF_Op<"TPUExecuteAndUpdateVariables", [DeclareOpInterfaceMe…
|