| /aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/tests/roundtrip/ |
| H A D | test11.pbtxt | 361 control_output: "AssignAddVariableOp"
453 name: "AssignAddVariableOp"
454 op: "AssignAddVariableOp"
464 original_node_names: "AssignAddVariableOp"
658 op: "AssignAddVariableOp"
689 input: "^AssignAddVariableOp"
700 key: "AssignAddVariableOp"
701 value: "AssignAddVariableOp"
724 control_output: "AssignAddVariableOp"
816 name: "AssignAddVariableOp"
[all …]
|
| H A D | graph-as-function-control-ret.pbtxt | 106 op: "AssignAddVariableOp"
139 op: "AssignAddVariableOp"
|
| H A D | test4.pbtxt | 1273 name: "cond_1/cond/AssignAddVariableOp/Switch"
1293 name: "cond_1/cond/AssignAddVariableOp"
1294 op: "AssignAddVariableOp"
1295 input: "cond_1/cond/AssignAddVariableOp/Switch:0"
1307 input: "cond_1/cond/AssignAddVariableOp/Switch:0"
1308 input: "^cond_1/cond/AssignAddVariableOp"
1320 input: "^cond_1/cond/AssignAddVariableOp"
|
| H A D | test10.pbtxt | 202 name: "AssignAddVariableOp"
203 op: "AssignAddVariableOp"
217 input: "^AssignAddVariableOp"
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| H A D | tf_saved_model_remove_vars_in_session_initializer.mlir | 49 // CHECK-NOT: tf.AssignAddVariableOp
57 …"tf.AssignAddVariableOp"(%0, %2) {dtype = "tfdtype$DT_INT32"} : (tensor<*x!tf_type.resource<tensor…
58 …"tf.AssignAddVariableOp"(%1, %3) {dtype = "tfdtype$DT_INT32"} : (tensor<*x!tf_type.resource<tensor…
70 // CHECK-NOT: tf.AssignAddVariableOp
79 …"tf.AssignAddVariableOp"(%0, %4) {dtype = "tfdtype$DT_INT32"} : (tensor<*x!tf_type.resource<tensor…
80 …"tf.AssignAddVariableOp"(%1, %4) {dtype = "tfdtype$DT_INT32"} : (tensor<*x!tf_type.resource<tensor…
|
| H A D | decompose_resource_ops.mlir | 14 // CLUSTER-ONLY: "tf.AssignAddVariableOp"
15 // ALWAYS-DECOMPOSE-NOT: "tf.AssignAddVariableOp"
16 …"tf.AssignAddVariableOp"(%0, %1) {dtype = "tfdtype$DT_INT32"} : (tensor<*x!tf_type.resource<tensor…
19 // CHECK-NOT: "tf.AssignAddVariableOp"
20 …"tf.AssignAddVariableOp"(%0, %1) {dtype = "tfdtype$DT_INT32"} : (tensor<*x!tf_type.resource<tensor…
46 // CHECK-NOT: "tf.AssignAddVariableOp"
47 …"tf.AssignAddVariableOp"(%arg0, %arg1) {dtype = "tfdtype$DT_INT32"} : (tensor<*x!tf_type.resource<…
54 // CHECK-NOT: "tf.AssignAddVariableOp"
55 …"tf.AssignAddVariableOp"(%arg0, %arg1) {dtype = "tfdtype$DT_INT32"} : (tensor<*x!tf_type.resource<…
61 // CLUSTER-ONLY: "tf.AssignAddVariableOp"
[all …]
|
| H A D | tpu_cluster_formation.mlir | 364 // CHECK: "tf.AssignAddVariableOp"([[USED_RESOURCE]], [[CONST]])
365 …"tf.AssignAddVariableOp"(%arg0, %0) : (tensor<*x!tf_type.resource<tensor<f32>>>, tensor<f32>) -> ()
379 // CHECK-NEXT: "tf.AssignAddVariableOp"([[UNUSED_RESOURCE]], [[CONST]])
387 "tf.AssignAddVariableOp"(%1, %0) : (tensor<*x!tf_type.resource<tensor<f32>>>, tensor<f32>) -> ()
445 "tf.AssignAddVariableOp"(%1, %0) : (!tf_res, tensor<f32>) -> ()
448 "tf.AssignAddVariableOp"(%1, %0) : (!tf_res, tensor<f32>) -> ()
458 // CHECK: "tf.AssignAddVariableOp"([[VAR]], [[CONST]])
464 // CHECK: "tf.AssignAddVariableOp"([[VAR]], [[CONST]])
473 …"tf.AssignAddVariableOp"(%1, %0) { _xla_compile_device_type = "TPU", _replication_info = "foo" } :…
475 "tf.AssignAddVariableOp"(%1, %0) : (!tf_res, tensor<f32>) -> ()
[all …]
|
| H A D | promote_var_handles_to_args.mlir | 51 // CHECK: "tf.AssignAddVariableOp"(%arg1, %arg0)
57 "tf.AssignAddVariableOp"(%2, %arg0) : (tensor<!tf_type.resource<tensor<f32>>>, tensor<f32>) -> ()
|
| H A D | tpu_parallel_execute_sink_resource_write.mlir | 29 // CHECK-NEXT: "tf.AssignAddVariableOp"([[ARG1]], [[PARALLEL_EXECUTE]]#0)
30 "tf.AssignAddVariableOp"(%arg1, %0#0) : (tensor<!tf_type.resource>, tensor<i1>) -> ()
|
| H A D | tf_saved_model_freeze_global_tensors.mlir | 116 … "tf.AssignAddVariableOp"(%arg0, %0) : (tensor<!tf_type.resource<tensor<f32>>>, tensor<f32>) -> ()
|
| H A D | tf_saved_model_optimize_global_tensors_interprocedural.mlir | 217 "tf.AssignAddVariableOp"(%arg0, %c0) : (tensor<*x!tf_type.resource>, tensor<f32>) -> ()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/tests/ |
| H A D | keras_imagenet_main.golden_summary | 6 AssignAddVariableOp 1
20 AssignAddVariableOp 1
51 AssignAddVariableOp 1
|
| H A D | keras_imagenet_main_graph_mode.golden_summary | 6 AssignAddVariableOp 2
18 AssignAddVariableOp 1
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
| H A D | graph-as-function-control-ret.pbtxt | 110 op: "AssignAddVariableOp"
143 op: "AssignAddVariableOp"
193 # CHECK-DAG: %[[VAR_ADD_1:.*]] = tf_executor.island wraps "tf.AssignAddVariableOp"(%[[ARG_1]], …
194 # CHECK-DAG: %[[VAR_ADD_2:.*]] = tf_executor.island wraps "tf.AssignAddVariableOp"(%[[ARG_2]], …
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| H A D | resource_spmd_expander.cc | 148 if (!llvm::isa<mlir::TF::AssignVariableOp, mlir::TF::AssignAddVariableOp, in ExpandOp()
234 if (llvm::isa<mlir::TF::AssignVariableOp, mlir::TF::AssignAddVariableOp, in ComputeLayoutForward()
252 if (llvm::isa<mlir::TF::AssignVariableOp, mlir::TF::AssignAddVariableOp, in ComputeLayoutBackward()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | variable_ops.cc | 106 class AssignAddVariableOp : public XlaOpKernel { class
108 explicit AssignAddVariableOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in AssignAddVariableOp() function in tensorflow::__anon735ed9830111::AssignAddVariableOp
121 AssignAddVariableOp);
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| H A D | analyze-variables.mlir | 52 …"tf.AssignAddVariableOp"(%0, %cst) {} : (tensor<*x!tf_type.resource<tensor<*xi32>>>, tensor<4xi32>…
95 …"tf.AssignAddVariableOp"(%arg4, %cst1) {} : (tensor<*x!tf_type.resource<tensor<*xi32>>>, tensor<4x…
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/api_def/java_api/ |
| H A D | api_def_AssignAddVariableOp.pbtxt | 2 graph_op_name: "AssignAddVariableOp"
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/api_def/python_api/ |
| H A D | api_def_AssignAddVariableOp.pbtxt | 2 graph_op_name: "AssignAddVariableOp"
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| H A D | AssignAddVariableOp.pbtxt | 2 name: "AssignAddVariableOp"
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| H A D | AssignAddVariableOp.pbtxt | 2 name: "AssignAddVariableOp"
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/ |
| H A D | api_def_AssignAddVariableOp.pbtxt | 2 graph_op_name: "AssignAddVariableOp"
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | tf_device_passes.td | 34 "tf.AssignAddVariableOp"(%resource_handle, %init_value)
205 tf.AssignAddVariableOp(%res, %0)
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | xla_compiler_test.cc | 387 auto write = ops::AssignAddVariableOp(scope, identity, a); in TEST_F()
441 auto write = ops::AssignAddVariableOp(scope, identity, a); in TEST_F()
505 auto write = ops::AssignAddVariableOp(scope, identity, a); in TEST_F()
1124 auto write = ops::AssignAddVariableOp(scope, identity, a); in TEST_F()
1283 auto write = ops::AssignAddVariableOp(scope, identity, a); in TEST_F()
1318 auto write = ops::AssignAddVariableOp(scope, var, a); in BuildTestGraph()
1886 auto write = ops::AssignAddVariableOp(scope, var, a); in TEST_F()
|
| /aosp_15_r20/external/tensorflow/tensorflow/go/op/ |
| H A D | scope_test.go | 79 update = AssignAddVariableOp(s, variable, one)
|