Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
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…
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)
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]])
Dtf-ops.mlir4157 …"tf.TPUExecuteAndUpdateVariables"(%arg0, %arg1) {device_var_reads_indices = [0, 1], device_var_upd…
4165 …"tf.TPUExecuteAndUpdateVariables"(%arg0, %arg1) {device_var_reads_indices = [0], device_var_update…
4173 …"tf.TPUExecuteAndUpdateVariables"(%arg0, %arg1) {device_var_reads_indices = [-1], device_var_updat…
4181 …"tf.TPUExecuteAndUpdateVariables"(%arg0, %arg1) {device_var_reads_indices = [0], device_var_update…
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DTPUExecuteAndUpdateVariables.pbtxt2 name: "TPUExecuteAndUpdateVariables"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_TPUExecuteAndUpdateVariables.pbtxt2 graph_op_name: "TPUExecuteAndUpdateVariables"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtf_passes.td991 `tf.TPUExecute` op and merges them into a `tf.TPUExecuteAndUpdateVariables`
1007 %2 = "tf.TPUExecuteAndUpdateVariables"(%arg0, %arg1, %compile)
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
Dlegalization.md3119 ### tf.TPUExecuteAndUpdateVariables
/external/tensorflow/tensorflow/go/op/
Dwrappers.go6879 func TPUExecuteAndUpdateVariables(scope *Scope, args []tf.Output, key tf.Output, Tresults []tf.Data… func
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt53433 name: "TPUExecuteAndUpdateVariables"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td16978 def TF_TPUExecuteAndUpdateVariablesOp : TF_Op<"TPUExecuteAndUpdateVariables", [DeclareOpInterfaceMe…