| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| 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.resource<tensor<2x8x…
58 …"tf.AssignAddVariableOp"(%1, %3) {dtype = "tfdtype$DT_INT32"} : (tensor<*x!tf.resource<tensor<2xi3…
70 // CHECK-NOT: tf.AssignAddVariableOp
79 …"tf.AssignAddVariableOp"(%0, %4) {dtype = "tfdtype$DT_INT32"} : (tensor<*x!tf.resource<tensor<2x8x…
80 …"tf.AssignAddVariableOp"(%1, %4) {dtype = "tfdtype$DT_INT32"} : (tensor<*x!tf.resource<tensor<2xi3…
|
| D | promote_var_handles_to_args.mlir | 51 // CHECK: "tf.AssignAddVariableOp"(%arg1, %arg0)
57 "tf.AssignAddVariableOp"(%2, %arg0) : (tensor<!tf.resource<tensor<f32>>>, tensor<f32>) -> ()
|
| D | tpu_cluster_formation.mlir | 399 // CHECK: "tf.AssignAddVariableOp"([[USED_RESOURCE]], [[CONST]])
400 "tf.AssignAddVariableOp"(%arg0, %0) : (tensor<*x!tf.resource<tensor<f32>>>, tensor<f32>) -> ()
414 // CHECK-NEXT: "tf.AssignAddVariableOp"([[UNUSED_RESOURCE]], [[CONST]])
422 "tf.AssignAddVariableOp"(%1, %0) : (tensor<*x!tf.resource<tensor<f32>>>, tensor<f32>) -> ()
480 "tf.AssignAddVariableOp"(%1, %0) : (!tf_res, tensor<f32>) -> ()
483 "tf.AssignAddVariableOp"(%1, %0) : (!tf_res, tensor<f32>) -> ()
493 // CHECK: "tf.AssignAddVariableOp"([[VAR]], [[CONST]])
499 // CHECK: "tf.AssignAddVariableOp"([[VAR]], [[CONST]])
508 "tf.AssignAddVariableOp"(%1, %0) { _tpu_replicate = "foo" } : (!tf_res, tensor<f32>) -> ()
510 "tf.AssignAddVariableOp"(%1, %0) : (!tf_res, tensor<f32>) -> ()
[all …]
|
| 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.resource>, tensor<i1>) -> ()
|
| D | tf_saved_model_optimize_global_tensors_interprocedural.mlir | 217 "tf.AssignAddVariableOp"(%arg0, %c0) : (tensor<*x!tf.resource>, tensor<f32>) -> ()
|
| D | decompose_resource_ops.mlir | 18 …"tf.AssignAddVariableOp"(%0, %1) {dtype = "tfdtype$DT_INT32"} : (tensor<*x!tf.resource<tensor<2x8x…
25 // Tests that composite tf.AssignAddVariableOp operation is decomposed and
39 …"tf.AssignAddVariableOp"(%0, %1) {dtype = "tfdtype$DT_INT32"} : (tensor<!tf.resource<tensor<i32>>>…
|
| D | tpu_space_to_depth_pass.mlir | 91 …ntrol_outputs = "IteratorGetNext,IteratorGetNext_1,CrossReplicaSum,AssignAddVariableOp,CrossReplic…
168 …ntrol_outputs = "IteratorGetNext,IteratorGetNext_1,CrossReplicaSum,AssignAddVariableOp,CrossReplic…
|
| /external/tensorflow/tensorflow/compiler/jit/tests/ |
| D | keras_imagenet_main.golden_summary | 6 AssignAddVariableOp 1
20 AssignAddVariableOp 1
51 AssignAddVariableOp 1
|
| D | keras_imagenet_main_graph_mode.golden_summary | 6 AssignAddVariableOp 2
18 AssignAddVariableOp 1
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
| 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]], …
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_AssignAddVariableOp.pbtxt | 2 graph_op_name: "AssignAddVariableOp"
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_AssignAddVariableOp.pbtxt | 2 graph_op_name: "AssignAddVariableOp"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | AssignAddVariableOp.pbtxt | 2 name: "AssignAddVariableOp"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | AssignAddVariableOp.pbtxt | 2 name: "AssignAddVariableOp"
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | variable_ops.cc | 92 class AssignAddVariableOp : public XlaOpKernel { class
94 explicit AssignAddVariableOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in AssignAddVariableOp() function in tensorflow::__anon798bfbf40111::AssignAddVariableOp
106 AssignAddVariableOp);
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_AssignAddVariableOp.pbtxt | 2 graph_op_name: "AssignAddVariableOp"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | scope_test.go | 79 update = AssignAddVariableOp(s, variable, one)
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler_test.cc | 381 auto write = ops::AssignAddVariableOp(scope, identity, a); in TEST_F()
433 auto write = ops::AssignAddVariableOp(scope, identity, a); in TEST_F()
497 auto write = ops::AssignAddVariableOp(scope, identity, a); in TEST_F()
1053 auto write = ops::AssignAddVariableOp(scope, identity, a); in TEST_F()
1208 auto write = ops::AssignAddVariableOp(scope, identity, a); in TEST_F()
1243 auto write = ops::AssignAddVariableOp(scope, var, a); in BuildTestGraph()
1863 auto write = ops::AssignAddVariableOp(scope, var, a); in TEST_F()
|
| D | functionalize_control_flow_test.cc | 1027 auto assign = ops::AssignAddVariableOp( in RunTest()
1278 auto assign = ops::AssignAddVariableOp( in CheckInnerNodesFunctionalized()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | resource_operation_safety_analysis_test.cc | 63 ops::AssignAddVariableOp assign_add_op(scope.WithOpName("Increment" + id), in MakeModify()
|
| D | mark_for_compilation_pass_test.cc | 1624 ops::AssignAddVariableOp increment_op( in TEST()
1628 ops::AssignAddVariableOp weights_0_update_op( in TEST()
1630 ops::AssignAddVariableOp weights_1_update_op( in TEST()
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | op_select_allowlist.md | 44 * `raw_ops.AssignAddVariableOp`
|
| /external/tensorflow/tensorflow/compiler/tf2xla/g3doc/ |
| D | gpu_supported_ops.md | 21 `AssignAddVariableOp` | `dtype={complex64,double,float,int32,int64,uint32,uint64}`
|
| D | cpu_supported_ops.md | 21 `AssignAddVariableOp` | `dtype={complex64,double,float,int32,int64,uint32,uint64}`
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 1263 ### tf.AssignAddVariableOp
1270 %output = tf.AssignAddVariableOp(%resource, %value, %dtype)
|