| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | tpu_parallel_execute_sink_resource_write.mlir | 13 // CHECK-NEXT: "tf.AssignVariableOp"([[ARG1]], [[PARALLEL_EXECUTE]]#0)
14 "tf.AssignVariableOp"(%arg1, %0#0) : (tensor<!tf.resource>, tensor<i1>) -> ()
44 // CHECK-NEXT: "tf.AssignVariableOp"([[PARALLEL_EXECUTE]]#0, [[ARG0]])
45 "tf.AssignVariableOp"(%0#0, %arg0) : (tensor<!tf.resource>, tensor<i1>) -> ()
57 // CHECK: "tf.AssignVariableOp"([[PARALLEL_EXECUTE]]#1, [[PARALLEL_EXECUTE]]#0)
58 "tf.AssignVariableOp"(%0#1, %0#0) : (tensor<!tf.resource>, tensor<i1>) -> ()
73 // CHECK-NEXT: "tf.AssignVariableOp"([[VAR]], [[PARALLEL_EXECUTE]]#0)
74 "tf.AssignVariableOp"(%1, %0#0) : (tensor<!tf.resource<tensor<i1>>>, tensor<i1>) -> ()
83 // CHECK-NEXT: "tf.AssignVariableOp"([[ARG3]], [[ARG1]])
92 "tf.AssignVariableOp"(%arg3, %0#1) : (tensor<!tf.resource>, tensor<i1>) -> ()
[all …]
|
| D | tpu-merge-variables-with-execute.mlir | 45 …"tf.AssignVariableOp"(%id0, %execute#0) : (tensor<*x!tf.resource<tensor<32xf32>>>, tensor<32xf32>)…
46 // CHECK-NEXT: "tf.AssignVariableOp"(%[[ARG_2]], %[[EXE]])
47 …"tf.AssignVariableOp"(%arg2, %execute#1) : (tensor<*x!tf.resource<tensor<16xf32>>>, tensor<16xf32>…
90 …"tf.AssignVariableOp"(%r, %execute) : (tensor<*x!tf.resource<tensor<32xf32>>>, tensor<32xf32>) -> …
122 // CHECK-NEXT: "tf.AssignVariableOp"(%[[ARG_0]], %[[ARG_2]])
123 …"tf.AssignVariableOp"(%arg0, %arg2) : (tensor<*x!tf.resource<tensor<32xf32>>>, tensor<32xf32>) -> …
124 // CHECK-NEXT: "tf.AssignVariableOp"(%[[ARG_5]], %[[ARG_6]])
125 "tf.AssignVariableOp"(%arg5, %arg6) : (tensor<*x!tf.resource<tensor<2xf32>>>, tensor<2xf32>) -> ()
153 …"tf.AssignVariableOp"(%arg1, %execute#1) : (tensor<*x!tf.resource<tensor<64xf32>>>, tensor<64xf32>…
154 // CHECK-NEXT: "tf.AssignVariableOp"(%[[ARG_0]], %[[EXE]]#0)
[all …]
|
| D | promote_resources_to_args.mlir | 25 // CHECK-NOT: "tf.AssignVariableOp"
29 "tf.AssignVariableOp"(%1, %0) : (tensor<!tf.resource<tensor<f32>>>, tensor<f32>) -> ()
84 // CHECK-NOT: "tf.AssignVariableOp"
95 "tf.AssignVariableOp"(%1, %4) : (tensor<!tf.resource<tensor<f32>>>, tensor<f32>) -> ()
107 // CHECK-NOT: "tf.AssignVariableOp"
116 "tf.AssignVariableOp"(%1, %0) : (tensor<!tf.resource<tensor<f32>>>, tensor<f32>) -> ()
120 "tf.AssignVariableOp"(%1, %4) : (tensor<!tf.resource<tensor<f32>>>, tensor<f32>) -> ()
181 "tf.AssignVariableOp"(%arg0, %0) : (tensor<!tf.resource<tensor<f32>>>, tensor<f32>) -> ()
196 "tf.AssignVariableOp"(%arg0, %0) : (tensor<!tf.resource<tensor<f32>>>, tensor<f32>) -> ()
199 "tf.AssignVariableOp"(%arg0, %1) : (tensor<!tf.resource<tensor<f32>>>, tensor<f32>) -> ()
[all …]
|
| D | stack_ops_decomposition.mlir | 16 // CHECK-NEXT: "tf.AssignVariableOp"(%[[SIZE]], %[[ZERO]])
17 // CHECK-NEXT: "tf.AssignVariableOp"(%[[BUFFER]], %[[BROADCAST]])
27 …// CHECK-NEXT: "tf.AssignVariableOp"(%[[BUFFER]], %[[UPDATE]]) : (tensor<!tf.resource<tensor<10xf3…
30 …// CHECK-NEXT: "tf.AssignVariableOp"(%[[SIZE]], %[[NEW_SIZE]]) : (tensor<!tf.resource<tensor<1xi32…
41 …// CHECK-NEXT: "tf.AssignVariableOp"(%[[SIZE]], %[[SUB]]) : (tensor<!tf.resource<tensor<1xi32>>>, …
61 …// CHECK-NEXT: "tf.AssignVariableOp"(%[[SIZE]], %[[ZERO_SIZE]]) : (tensor<!tf.resource<tensor<1xi3…
62 …// CHECK-NEXT: "tf.AssignVariableOp"(%[[BUFFER]], %[[BROADCAST]]) : (tensor<!tf.resource<tensor<10…
74 …// CHECK-NEXT: "tf.AssignVariableOp"(%[[BUFFER]], %[[UPDATE]]) : (tensor<!tf.resource<tensor<10x2x…
77 …// CHECK-NEXT: "tf.AssignVariableOp"(%[[SIZE]], %[[NEW_SIZE]]) : (tensor<!tf.resource<tensor<1xi32…
133 // CHECK: tf.AssignVariableOp
[all …]
|
| D | tpu-resource-read-for-write.mlir | 10 // CHECK-NEXT: "tf.AssignVariableOp"([[ARG2]], [[CLUSTER]]#1)
11 "tf.AssignVariableOp"(%arg2, %0#1) : (tensor<*x!tf.resource<tensor<i32>>>, tensor<i32>) -> ()
27 "tf.AssignVariableOp"(%arg2, %1#1) : (tensor<*x!tf.resource<tensor<i32>>>, tensor<i32>) -> ()
41 "tf.AssignVariableOp"(%arg1, %0#0) : (tensor<*x!tf.resource<tensor<i32>>>, tensor<i32>) -> ()
42 "tf.AssignVariableOp"(%arg1, %0#1) : (tensor<*x!tf.resource<tensor<i32>>>, tensor<i32>) -> ()
56 "tf.AssignVariableOp"(%arg1, %0) : (tensor<*x!tf.resource<tensor<i32>>>, tensor<i32>) -> ()
|
| D | decompose_resource_ops.mlir | 15 // CHECK: "tf.AssignVariableOp"
36 // CHECK: "tf.AssignVariableOp"
57 // CHECK: "tf.AssignVariableOp"
80 // CHECK: "tf.AssignVariableOp"(%[[RES_HANDLE]], %[[SUB]])
105 // CHECK: "tf.AssignVariableOp"([[ACCUM_HANDLE]], [[ACCUM_NEW]])
109 // CHECK: "tf.AssignVariableOp"([[VAR_HANDLE]], [[VAR_NEW]])
131 // CHECK: "tf.AssignVariableOp"([[ACCUM_HANDLE]], [[ACCUM_NEW]])
138 // CHECK: "tf.AssignVariableOp"([[VAR_HANDLE]], [[VAR_NEW]])
161 // CHECK: "tf.AssignVariableOp"(%[[ACCUM_HANDLE]], %[[NEW_ACCUM]])
165 // CHECK: "tf.AssignVariableOp"(%[[VAR_HANDLE]], %[[NEW_VAR]])
[all …]
|
| D | tpu_resource_partitioning.mlir | 20 // CHECK-DAG: "tf.AssignVariableOp"([[ARG0]], [[OUTPUT]]#0)
21 // CHECK-DAG: "tf.AssignVariableOp"([[ARG1]], [[OUTPUT]]#1)
22 "tf.AssignVariableOp"(%0, %2) : (tensor<!tf.resource<tensor<i32>>>, tensor<i32>) -> ()
39 // CHECK-NOT: tf.AssignVariableOp
65 "tf.AssignVariableOp"(%arg0, %1) : (tensor<!tf.resource<tensor<i32>>>, tensor<i32>) -> ()
75 // CHECK-NOT: tf.AssignVariableOp
115 "tf.AssignVariableOp"(%0, %2) : (tensor<!tf.resource>, tensor<i32>) -> ()
|
| D | resource_op_lifting.mlir | 42 // CHECK: "tf.AssignVariableOp"(%[[RES_HANDLE]], %[[CLUSTER_RES]]#1)
46 …"tf.AssignVariableOp"(%0, %2) {dtype = i32} : (tensor<*x!tf.resource<tensor<*xi32>>>, tensor<*xi32…
70 // CHECK: "tf.AssignVariableOp"(%[[RES_HANDLE]], %[[CLUSTER_RES]]#1)
75 …"tf.AssignVariableOp"(%0, %3) {dtype = i32} : (tensor<*x!tf.resource<tensor<*xi32>>>, tensor<*xi32…
101 // CHECK: "tf.AssignVariableOp"(%[[RES_HANDLE]], %[[CLUSTER_RES]]#1)
106 …"tf.AssignVariableOp"(%0, %3) {dtype = i32} : (tensor<*x!tf.resource<tensor<*xi32>>>, tensor<*xi32…
134 // CHECK: "tf.AssignVariableOp"(%[[RES_HANDLE]], %[[COMPUTE_RES]])
135 …"tf.AssignVariableOp"(%1, %3) {dtype = i32} : (tensor<*x!tf.resource<tensor<*xi32>>>, tensor<*xi32…
169 // CHECK: "tf.AssignVariableOp"(%[[VH]], %[[CLUSTER]])
179 "tf.AssignVariableOp"(%arg1, %add0) : (tensor<*x!tf.resource<tensor<f32>>>, tensor<f32>) -> ()
[all …]
|
| D | tensor_array_ops_decomposition.mlir | 11 // CHECK: "tf.AssignVariableOp"(%[[VAR]], %[[BUFFER]])
21 // CHECK: "tf.AssignVariableOp"(%[[VAR]], %[[NEW_BUFFER]])
44 // CHECK: "tf.AssignVariableOp"(%[[VAR]], %[[BUFFER]])
108 // CHECK: "tf.AssignVariableOp"
119 // CHECK: "tf.AssignVariableOp"(%[[VAR]], %[[ADD]])
132 // CHECK: "tf.AssignVariableOp"
141 // CHECK: "tf.AssignVariableOp"(%[[VAR]], %[[ADD]])
154 // CHECK: "tf.AssignVariableOp"
187 // CHECK: "tf.AssignVariableOp"(%[[VAR]], %[[UPDATE1]])
200 // CHECK: "tf.AssignVariableOp"(%[[VAR]],
[all …]
|
| D | breakup-islands.mlir | 177 …"tf.AssignVariableOp"(%arg0, %arg2) : (tensor<*x!tf.resource<tensor<32xf32>>>, tensor<32xf32>) -> …
181 …"tf.AssignVariableOp"(%arg1, %read0) : (tensor<*x!tf.resource<tensor<32xf32>>>, tensor<32xf32>) ->…
182 …"tf.AssignVariableOp"(%arg0, %read2) : (tensor<*x!tf.resource<tensor<32xf32>>>, tensor<32xf32>) ->…
194 …GN0_CONTROL:.*]] = tf_executor.island(%[[READ0_CONTROL]]) wraps "tf.AssignVariableOp"(%arg0, %arg2)
198 …GN1_CONTROL:.*]] = tf_executor.island(%[[READ1_CONTROL]]) wraps "tf.AssignVariableOp"(%arg1, %[[RE…
199 …2_CONTROL:.*]] = tf_executor.island(%[[ASSIGN0_CONTROL]]) wraps "tf.AssignVariableOp"(%arg0, %[[RE…
211 …"tf.AssignVariableOp"(%vh1, %arg0) : (tensor<*x!tf.resource<tensor<32xf32>>>, tensor<32xf32>) -> ()
214 …"tf.AssignVariableOp"(%vh0, %read1) : (tensor<*x!tf.resource<tensor<32xf32>>>, tensor<32xf32>) -> …
215 …"tf.AssignVariableOp"(%vh1, %read0) : (tensor<*x!tf.resource<tensor<32xf32>>>, tensor<32xf32>) -> …
228 // CHECK: %[[ASSIGN0_CONTROL:.*]] = tf_executor.island wraps "tf.AssignVariableOp"(%[[VH1]], %arg…
[all …]
|
| D | tpu_colocate_composite_resource_ops.mlir | 80 // CHECK-NEXT: "tf.AssignVariableOp"(%[[RI_0]], %[[VAL_OUT]])
84 "tf.AssignVariableOp"(%arg1, %1) : (tensor<*x!tf.resource<tensor<4xf32>>>, tensor<4xf32>) -> ()
107 // CHECK-NEXT: "tf.AssignVariableOp"(%[[RI_0]], %[[VAL_OUT]])
109 "tf.AssignVariableOp"(%arg1, %1) : (tensor<*x!tf.resource<tensor<4xf32>>>, tensor<4xf32>) -> ()
|
| D | side-effect-analysis-test.mlir | 22 …"tf.AssignVariableOp"(%arg0, %arg2) : (tensor<*x!tf.resource<tensor<32xf32>>>, tensor<32xf32>) -> …
34 …"tf.AssignVariableOp"(%arg1, %read0) : (tensor<*x!tf.resource<tensor<32xf32>>>, tensor<32xf32>) ->…
38 …"tf.AssignVariableOp"(%arg0, %read2) : (tensor<*x!tf.resource<tensor<32xf32>>>, tensor<32xf32>) ->…
83 …"tf.AssignVariableOp"(%vh1_id#0, %arg0) : (tensor<*x!tf.resource<tensor<32xf32>>>, tensor<32xf32>)…
95 …"tf.AssignVariableOp"(%vh0, %read2) : (tensor<*x!tf.resource<tensor<32xf32>>>, tensor<32xf32>) -> …
99 …"tf.AssignVariableOp"(%vh1_id#0, %read1) : (tensor<*x!tf.resource<tensor<32xf32>>>, tensor<32xf32>…
138 …"tf.AssignVariableOp"(%vh1, %arg0) : (tensor<*x!tf.resource<tensor<32xf32>>>, tensor<32xf32>) -> ()
153 …"tf.AssignVariableOp"(%vh0, %read1) : (tensor<*x!tf.resource<tensor<32xf32>>>, tensor<32xf32>) -> …
157 …"tf.AssignVariableOp"(%vh1, %read0) : (tensor<*x!tf.resource<tensor<32xf32>>>, tensor<32xf32>) -> …
202 …"tf.AssignVariableOp"(%vh1, %read0) : (tensor<*x!tf.resource<tensor<32xf32>>>, tensor<32xf32>) -> …
[all …]
|
| D | resource-alias-analysis-test.mlir | 17 "tf.AssignVariableOp"(%arg0, %arg2) : (!tf_res, tensor<32xf32>) -> ()
22 "tf.AssignVariableOp"(%arg1, %read0) : (!tf_res, tensor<32xf32>) -> ()
23 "tf.AssignVariableOp"(%arg0, %read2) : (!tf_res, tensor<32xf32>) -> ()
48 "tf.AssignVariableOp"(%vh1_id#0, %arg0) : (!tf_res, tensor<32xf32>) -> ()
51 "tf.AssignVariableOp"(%vh0, %read2) : (!tf_res, tensor<32xf32>) -> ()
52 "tf.AssignVariableOp"(%vh1_id#0, %read1) : (!tf_res, tensor<32xf32>) -> ()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_variables.cc | 71 TF::AssignVariableOp assign_op, in GetAssignVariableOp()
73 return builder.create<TFL::AssignVariableOp>( in GetAssignVariableOp()
107 if (llvm::isa<TF::AssignVariableOp>(op)) { in matchAndRewrite()
108 auto assign_op = llvm::cast<TF::AssignVariableOp>(op); in matchAndRewrite()
149 LegalizeVariablesPattern<TF::AssignVariableOp>>( in runOnOperation()
|
| D | initialize_variables.cc | 83 builder.create<TFL::AssignVariableOp>(session_init.body().getLoc(), in InitializeVariable()
92 else if (auto assign_var = llvm::dyn_cast_or_null<TF::AssignVariableOp>(op)) in GetGlobalTensorOp()
93 return GetGlobalTensor<TF::AssignVariableOp>(symbol_table, assign_var, in GetGlobalTensorOp()
124 if (!llvm::isa<TF::ReadVariableOp, TF::AssignVariableOp>(op)) in InitializeVariables()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_parallel_execute_sink_resource_write.cc | 48 TF::AssignVariableOp GetSingleUseResourceWrite( in GetSingleUseResourceWrite()
53 auto assign_var = dyn_cast<TF::AssignVariableOp>(use.getOwner()); in GetSingleUseResourceWrite()
91 TF::AssignVariableOp assign_var = in SinkResourceWritesIntoParallelExecute()
|
| D | tpu_resource_read_for_write.cc | 53 auto assign_var = dyn_cast<TF::AssignVariableOp>(result_user); in GetResourceWriteResult()
60 auto assign_var_user = dyn_cast<TF::AssignVariableOp>(handle_user); in GetResourceWriteResult()
|
| D | tpu_resource_partitioning.cc | 90 llvm::dyn_cast<TF::AssignVariableOp>(*result.getUsers().begin()); in PartitionResourceReadsWrites()
109 builder.create<TF::AssignVariableOp>( in PartitionResourceReadsWrites()
|
| /external/tensorflow/tensorflow/cc/saved_model/testdata/CyclicModule/debug/ |
| D | saved_model_debug_info.pb | 5 :AssignVariableOp/MyVariable@__inference__traced_restore_45$
12 /AssignVariableOp@__inference__traced_restore_45$
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_AssignVariableOp.pbtxt | 2 graph_op_name: "AssignVariableOp"
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_AssignVariableOp.pbtxt | 2 graph_op_name: "AssignVariableOp"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | AssignVariableOp.pbtxt | 2 name: "AssignVariableOp"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | AssignVariableOp.pbtxt | 2 name: "AssignVariableOp"
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | variable_ops.cc | 82 class AssignVariableOp : public XlaOpKernel { class
84 explicit AssignVariableOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in AssignVariableOp() function in tensorflow::__anon798bfbf40111::AssignVariableOp
90 REGISTER_XLA_OP(Name("AssignVariableOp").CompilationOnly(), AssignVariableOp);
|
| /external/tensorflow/tensorflow/cc/saved_model/testdata/VarsAndArithmeticObjectGraph/debug/ |
| D | saved_model_debug_info.pb | 11 ;AssignVariableOp/variable_x@__inference__traced_restore_101(
18 0AssignVariableOp@__inference__traced_restore_101(
|