/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)
|