AssignVariableOp (reference) in projects: external

Project(s)

Full Search
Definition
Symbol
File Path
History
Type

Searched refs:AssignVariableOp (Results 1 – 25 of 96) sorted by relevance

12 3 4

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
D	hoist_replicate_invariant_resource_writes.mlir	15 // CHECK-NOT: tf.AssignVariableOp 16 "tf.AssignVariableOp"(%arg0, %op_a) : (!tf_res_f32, tensor<f32>) -> () 20 // CHECK: "tf.AssignVariableOp"([[ARG0]], [[REPLICATE]]#2) 31 // CHECK: "tf.AssignVariableOp"([[ARG0]], [[OP_A]]) 32 "tf.AssignVariableOp"(%arg0, %op_a) : (!tf_res_f32, tensor<f32>) -> () 38 // CHECK-NOT: tf.AssignVariableOp 52 // CHECK: "tf.AssignVariableOp"([[RI]], [[OP_A]]) 53 "tf.AssignVariableOp"(%ri, %op_a) : (!tf_res_f32, tensor<f32>) -> () 57 // CHECK-NOT: tf.AssignVariableOp 70 // CHECK: "tf.AssignVariableOp"([[ARG0]], [[OP_A]]) [all …]
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_type.resource>, tensor<i1>) -> () 44 // CHECK-NEXT: "tf.AssignVariableOp"([[PARALLEL_EXECUTE]]#0, [[ARG0]]) 45 "tf.AssignVariableOp"(%0#0, %arg0) : (tensor<!tf_type.resource>, tensor<i1>) -> () 57 // CHECK: "tf.AssignVariableOp"([[PARALLEL_EXECUTE]]#1, [[PARALLEL_EXECUTE]]#0) 58 "tf.AssignVariableOp"(%0#1, %0#0) : (tensor<!tf_type.resource>, tensor<i1>) -> () 73 // CHECK-NEXT: "tf.AssignVariableOp"([[VAR]], [[PARALLEL_EXECUTE]]#0) 74 "tf.AssignVariableOp"(%1, %0#0) : (tensor<!tf_type.resource<tensor<i1>>>, tensor<i1>) -> () 83 // CHECK-NEXT: "tf.AssignVariableOp"([[ARG3]], [[ARG1]]) 92 "tf.AssignVariableOp"(%arg3, %0#1) : (tensor<!tf_type.resource>, tensor<i1>) -> () [all …]
D	mark_input_output_aliases.mlir	6 // AssignVariableOp. If a pair of input-output (say input at index `a` and 21 "tf.AssignVariableOp"(%arg1, %device_output#0) : (!tf_res_f32, tensor<f32>) -> () 22 "tf.AssignVariableOp"(%arg0, %device_output#1) : (!tf_res_i32, tensor<i32>) -> () 38 "tf.AssignVariableOp"(%arg0, %device_output) : (!tf_res_i32, tensor<i32>) -> () 53 "tf.AssignVariableOp"(%arg0, %device_output) : (!tf_res_i32, tensor<i32>) -> () 69 "tf.AssignVariableOp"(%arg0, %device_output#0) : (!tf_res_f32, tensor<f32>) -> () 70 "tf.AssignVariableOp"(%arg0, %device_output#1) : (!tf_res_f32, tensor<f32>) -> () 85 "tf.AssignVariableOp"(%arg0, %device_output#0) : (!tf_res_f32, tensor<f32>) -> () 86 "tf.AssignVariableOp"(%arg1, %device_output#1) : (!tf_res_f32, tensor<f32>) -> ()
D	tpu-merge-variables-with-execute.mlir	42 …"tf.AssignVariableOp"(%id0, %execute#0) : (tensor<x!tf_type.resource<tensor<32xf32>>>, tensor<32x… 43 // CHECK-NEXT: "tf.AssignVariableOp"(%[[ARG_2]], %[[EXE]]) 44 …"tf.AssignVariableOp"(%arg2, %execute#1) : (tensor<x!tf_type.resource<tensor<16xf32>>>, tensor<16… 85 …"tf.AssignVariableOp"(%r, %execute) : (tensor<x!tf_type.resource<tensor<32xf32>>>, tensor<32xf32>… 117 // CHECK-NEXT: "tf.AssignVariableOp"(%[[ARG_0]], %[[ARG_2]]) 118 …"tf.AssignVariableOp"(%arg0, %arg2) : (tensor<x!tf_type.resource<tensor<32xf32>>>, tensor<32xf32>… 119 // CHECK-NEXT: "tf.AssignVariableOp"(%[[ARG_5]], %[[ARG_6]]) 120 …"tf.AssignVariableOp"(%arg5, %arg6) : (tensor<x!tf_type.resource<tensor<2xf32>>>, tensor<2xf32>) … 146 …"tf.AssignVariableOp"(%arg1, %execute#1) : (tensor<x!tf_type.resource<tensor<64xf32>>>, tensor<64… 147 // CHECK-NEXT: "tf.AssignVariableOp"(%[[ARG_0]], %[[EXE]]#0) [all …]
D	promote_resources_to_args.mlir	49 // CHECK-NOT: "tf.AssignVariableOp" 54 "tf.AssignVariableOp"(%1, %0) : (tensor<!tf_type.resource<tensor<f32>>>, tensor<f32>) -> () 111 // CHECK-NOT: "tf.AssignVariableOp" 123 "tf.AssignVariableOp"(%1, %4) : (tensor<!tf_type.resource<tensor<f32>>>, tensor<f32>) -> () 135 // CHECK-NOT: "tf.AssignVariableOp" 144 "tf.AssignVariableOp"(%1, %0) : (tensor<!tf_type.resource<tensor<f32>>>, tensor<f32>) -> () 148 "tf.AssignVariableOp"(%1, %4) : (tensor<!tf_type.resource<tensor<f32>>>, tensor<f32>) -> () 209 "tf.AssignVariableOp"(%arg0, %0) : (tensor<!tf_type.resource<tensor<f32>>>, tensor<f32>) -> () 224 "tf.AssignVariableOp"(%arg0, %0) : (tensor<!tf_type.resource<tensor<f32>>>, tensor<f32>) -> () 227 "tf.AssignVariableOp"(%arg0, %1) : (tensor<!tf_type.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_type.resource<tensor<… 30 …// CHECK-NEXT: "tf.AssignVariableOp"(%[[SIZE]], %[[NEW_SIZE]]) : (tensor<!tf_type.resource<tensor<… 41 …// CHECK-NEXT: "tf.AssignVariableOp"(%[[SIZE]], %[[SUB]]) : (tensor<!tf_type.resource<tensor<1xi32… 61 …// CHECK-NEXT: "tf.AssignVariableOp"(%[[SIZE]], %[[ZERO_SIZE]]) : (tensor<!tf_type.resource<tensor… 62 …// CHECK-NEXT: "tf.AssignVariableOp"(%[[BUFFER]], %[[BROADCAST]]) : (tensor<!tf_type.resource<tens… 74 …// CHECK-NEXT: "tf.AssignVariableOp"(%[[BUFFER]], %[[UPDATE]]) : (tensor<!tf_type.resource<tensor<… 77 …// CHECK-NEXT: "tf.AssignVariableOp"(%[[SIZE]], %[[NEW_SIZE]]) : (tensor<!tf_type.resource<tensor<… 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_type.resource<tensor<i32>>>, tensor<i32>) -> () 27 "tf.AssignVariableOp"(%arg2, %1#1) : (tensor<x!tf_type.resource<tensor<i32>>>, tensor<i32>) -> () 41 "tf.AssignVariableOp"(%arg1, %0#0) : (tensor<x!tf_type.resource<tensor<i32>>>, tensor<i32>) -> () 42 "tf.AssignVariableOp"(%arg1, %0#1) : (tensor<x!tf_type.resource<tensor<i32>>>, tensor<i32>) -> () 56 "tf.AssignVariableOp"(%arg1, %0) : (tensor<*x!tf_type.resource<tensor<i32>>>, 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_type.resource<tensor<xi32>>>, tensor<… 70 // CHECK: "tf.AssignVariableOp"(%[[RES_HANDLE]], %[[CLUSTER_RES]]#1) 75 …"tf.AssignVariableOp"(%0, %3) {dtype = i32} : (tensor<x!tf_type.resource<tensor<xi32>>>, tensor<… 101 // CHECK: "tf.AssignVariableOp"(%[[RES_HANDLE]], %[[CLUSTER_RES]]#1) 106 …"tf.AssignVariableOp"(%0, %3) {dtype = i32} : (tensor<x!tf_type.resource<tensor<xi32>>>, tensor<… 134 // CHECK: "tf.AssignVariableOp"(%[[RES_HANDLE]], %[[COMPUTE_RES]]) 135 …"tf.AssignVariableOp"(%1, %3) {dtype = i32} : (tensor<x!tf_type.resource<tensor<xi32>>>, tensor<… 169 // CHECK: "tf.AssignVariableOp"(%[[VH]], %[[CLUSTER]]) 179 …"tf.AssignVariableOp"(%arg1, %add0) : (tensor<*x!tf_type.resource<tensor<f32>>>, tensor<f32>) -> () [all …]
D	decompose_resource_ops.mlir	82 // CHECK: "tf.AssignVariableOp" 104 // CHECK: "tf.AssignVariableOp" 127 // CHECK: "tf.AssignVariableOp" 152 // CHECK: "tf.AssignVariableOp"(%[[RES_HANDLE]], %[[SUB]]) 179 // CHECK: "tf.AssignVariableOp"([[ACCUM_HANDLE]], [[ACCUM_NEW]]) 183 // CHECK: "tf.AssignVariableOp"([[VAR_HANDLE]], [[VAR_NEW]]) 207 // CHECK: "tf.AssignVariableOp"([[ACCUM_HANDLE]], [[ACCUM_NEW]]) 214 // CHECK: "tf.AssignVariableOp"([[VAR_HANDLE]], [[VAR_NEW]]) 260 // CHECK: "tf.AssignVariableOp"(%[[ACCUM_HANDLE]], %[[NEW_ACCUM]]) 264 // 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_type.resource<tensor<i32>>>, tensor<i32>) -> () 39 // CHECK-NOT: tf.AssignVariableOp 84 "tf.AssignVariableOp"(%arg0, %1) : (tensor<!tf_type.resource<tensor<i32>>>, tensor<i32>) -> () 94 // CHECK-NOT: tf.AssignVariableOp 124 "tf.AssignVariableOp"(%0, %2) : (tensor<!tf_type.resource>, tensor<i32>) -> ()`
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	tf_saved_model_initialize_variables_in_session_init.mlir	`19 // CHECK: "tf.AssignVariableOp"(%[[VAR]], %[[CST]]) 44 // CHECK: "tf.AssignVariableOp"(%[[VAR]], %[[CST]]) 80 // CHECK: "tf.AssignVariableOp"(%[[VAR]], %[[CST]]) 87 // CHECK-NOT: "tf.AssignVariableOp"()`
D	breakup-islands.mlir	177 …"tf.AssignVariableOp"(%arg0, %arg2) : (tensor<x!tf_type.resource<tensor<32xf32>>>, tensor<32xf32>… 181 …"tf.AssignVariableOp"(%arg1, %read0) : (tensor<x!tf_type.resource<tensor<32xf32>>>, tensor<32xf32… 182 …"tf.AssignVariableOp"(%arg0, %read2) : (tensor<x!tf_type.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_type.resource<tensor<32xf32>>>, tensor<32xf32>)… 214 …"tf.AssignVariableOp"(%vh0, %read1) : (tensor<x!tf_type.resource<tensor<32xf32>>>, tensor<32xf32>… 215 …"tf.AssignVariableOp"(%vh1, %read0) : (tensor<x!tf_type.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_type.resource<tensor<4xf32>>>, tensor<4xf32>) -> … 107 // CHECK-NEXT: "tf.AssignVariableOp"(%[[RI_0]], %[[VAL_OUT]]) 109 …"tf.AssignVariableOp"(%arg1, %1) : (tensor<x!tf_type.resource<tensor<4xf32>>>, tensor<4xf32>) -> …`
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
D	hoist_replicate_invariant_resource_writes.cc	60 llvm::ArrayRef<TF::AssignVariableOp> tail_assign_variable_ops) { in MoveTailWritesAfterReplicate() 103 SmallVector<TF::AssignVariableOp> GetTailWritesToReplicateInvariantResourceVars( in GetTailWritesToReplicateInvariantResourceVars() 105 SmallVector<TF::AssignVariableOp, 16> tail_assign_variable_ops; in GetTailWritesToReplicateInvariantResourceVars() 112 if (auto assign = llvm::dyn_cast<TF::AssignVariableOp>(op)) { in GetTailWritesToReplicateInvariantResourceVars() 133 SmallVector<TF::AssignVariableOp> tail_writes = in runOnFunction()
D	tpu_parallel_execute_sink_resource_write.cc	`57 TF::AssignVariableOp GetSingleUseResourceWrite( in GetSingleUseResourceWrite() 62 auto assign_var = dyn_cast<TF::AssignVariableOp>(use.getOwner()); in GetSingleUseResourceWrite() 100 TF::AssignVariableOp assign_var = in SinkResourceWritesIntoParallelExecute()`
D	tf_device_passes.td	`49 "tf.AssignVariableOp"(%resource_handle, %1#1) 77 "tf.AssignVariableOp"(%arg0, %constant) 96 "tf.AssignVariableOp"(%0, %2) 195 …ecompose composite resource variable operations into primitive Read/AssignVariableOp and raw compu… 199 ReadVariableOp, AssignVariableOp and other computations to facilitate 213 tf.AssignVariableOp(%res, %1) 225 …ecompose composite resource variable operations into primitive Read/AssignVariableOp and raw compu…`
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/compiler/mlir/tfrt/tests/tf_to_corert/
D	lower_saved_model_error.mlir	`16 // expected-error @+1 {{'tf.AssignVariableOp' op unsupported users of resource variables}} 17 …"tf.AssignVariableOp"(%arg0, %0) {device = "/device:CPU:0", dtype = i32} : (tensor<!tf_type.resour… 50 // expected-error @+1 {{'tf.AssignVariableOp' op unsupported users of resource variables}} 51 …"tf.AssignVariableOp"(%res, %0) {device = "/device:CPU:0", dtype = i32} : (tensor<!tf_type.resourc…`
D	side_effects.mlir	`7 …async.executeop.seq([[in_chain]]) key(1) cost({{.*}}) device("/device:CPU:0") "tf.AssignVariableOp" 13 …"tf.AssignVariableOp"(%1, %0) {device = "/device:CPU:0"} : (tensor<!tf_type.resource<tensor<f32>>>…`
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
D	legalize-tf-variables.mlir	`12 …"tf.AssignVariableOp"(%handle_0, %1) : (tensor<!tf_type.resource<tensor<1x10xf32>>>, tensor<1x10xf… 41 …"tf.AssignVariableOp"(%handle_0, %cst_1) : (tensor<!tf_type.resource<tensor<1x10xf32>>>, tensor<1x… 57 …"tf.AssignVariableOp"(%handle_0, %1) : (tensor<!tf_type.resource<tensor<1x10xf32>>>, tensor<1x10xf… 91 …"tf.AssignVariableOp"(%handle_0, %cst) : (tensor<!tf_type.resource<tensor<1x10xui64>>>, tensor<1x1… 96 // CHECK: tf.AssignVariableOp`
/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/compiler/mlir/tfrt/tests/
D	convert_ref_variables.mlir	`12 // CHECK: "tf.AssignVariableOp"([[handle]], {{%.*}}) 45 // CHECK-NEXT: "tf.AssignVariableOp"([[handle]], [[zero]]) 53 // CHECK-NEXT: "tf.AssignVariableOp"([[handle]], [[one]])`
/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/analysis/
D	compatibility_analysis.mlir	`13 // CHECK-NEXT: key: "tf.AssignVariableOp" 57 "tf.AssignVariableOp"(%2, %1) : (tensor<*x!tf_type.resource<tensor<f32>>>, tensor<f32>) -> ()`

12 3 4

art
bionic
bootable
build
cts
dalvik
developers
development
device
external
frameworks
hardware
kernel
libcore
libnativehelper
packages
pdk
platform_testing
sdk
system
test
toolchain
tools