Home
last modified time | relevance | path

Searched refs:AssignAddVariableOp (Results 1 – 25 of 42) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/tests/roundtrip/
H A Dtest11.pbtxt361 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 Dgraph-as-function-control-ret.pbtxt106 op: "AssignAddVariableOp"
139 op: "AssignAddVariableOp"
H A Dtest4.pbtxt1273 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 Dtest10.pbtxt202 name: "AssignAddVariableOp"
203 op: "AssignAddVariableOp"
217 input: "^AssignAddVariableOp"
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
H A Dtf_saved_model_remove_vars_in_session_initializer.mlir49 // 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 Ddecompose_resource_ops.mlir14 // 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 Dtpu_cluster_formation.mlir364 // 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 Dpromote_var_handles_to_args.mlir51 // CHECK: "tf.AssignAddVariableOp"(%arg1, %arg0)
57 "tf.AssignAddVariableOp"(%2, %arg0) : (tensor<!tf_type.resource<tensor<f32>>>, tensor<f32>) -> ()
H A Dtpu_parallel_execute_sink_resource_write.mlir29 // CHECK-NEXT: "tf.AssignAddVariableOp"([[ARG1]], [[PARALLEL_EXECUTE]]#0)
30 "tf.AssignAddVariableOp"(%arg1, %0#0) : (tensor<!tf_type.resource>, tensor<i1>) -> ()
H A Dtf_saved_model_freeze_global_tensors.mlir116 … "tf.AssignAddVariableOp"(%arg0, %0) : (tensor<!tf_type.resource<tensor<f32>>>, tensor<f32>) -> ()
H A Dtf_saved_model_optimize_global_tensors_interprocedural.mlir217 "tf.AssignAddVariableOp"(%arg0, %c0) : (tensor<*x!tf_type.resource>, tensor<f32>) -> ()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/tests/
H A Dkeras_imagenet_main.golden_summary6 AssignAddVariableOp 1
20 AssignAddVariableOp 1
51 AssignAddVariableOp 1
H A Dkeras_imagenet_main_graph_mode.golden_summary6 AssignAddVariableOp 2
18 AssignAddVariableOp 1
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/
H A Dgraph-as-function-control-ret.pbtxt110 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 Dresource_spmd_expander.cc148 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 Dvariable_ops.cc106 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 Danalyze-variables.mlir52 …"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 Dapi_def_AssignAddVariableOp.pbtxt2 graph_op_name: "AssignAddVariableOp"
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/python_api/
H A Dapi_def_AssignAddVariableOp.pbtxt2 graph_op_name: "AssignAddVariableOp"
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
H A DAssignAddVariableOp.pbtxt2 name: "AssignAddVariableOp"
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
H A DAssignAddVariableOp.pbtxt2 name: "AssignAddVariableOp"
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/
H A Dapi_def_AssignAddVariableOp.pbtxt2 graph_op_name: "AssignAddVariableOp"
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtf_device_passes.td34 "tf.AssignAddVariableOp"(%resource_handle, %init_value)
205 tf.AssignAddVariableOp(%res, %0)
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_compiler_test.cc387 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 Dscope_test.go79 update = AssignAddVariableOp(s, variable, one)

12