Searched refs:assign_variable (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | assign_variable.cc | 27 namespace assign_variable { namespace
75 static TfLiteRegistration r = {nullptr, nullptr, assign_variable::Prepare, in Register_ASSIGN_VARIABLE()
76 assign_variable::Eval}; in Register_ASSIGN_VARIABLE()
|
| D | BUILD | 704 "assign_variable.cc",
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/compile_mlir_util/ |
| D | graph-resource.pbtxt | 1 …e-on-import=false %s -tf-graph-as-function -tf-control-output-arrays=assign_variable | tf-mlir-tra…
36 name: "assign_variable"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | optimize_global_tensors.cc | 100 if (auto assign_variable = dyn_cast<TF::AssignVariableOp>(op)) { in AnalyzeFunc() local
101 SetPotentiallyWritten(assign_variable.resource()); in AnalyzeFunc()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | initialize_variables.mlir | 18 // CHECK: "tfl.assign_variable"(%[[RESOURCE]], %[[VAL]]) : (tensor<1xi32>, tensor<1x10xf32>) -> ()
40 … // CHECK: "tfl.assign_variable"(%[[RESOURCE]], %[[VAL]]) : (tensor<1xi32>, tensor<1x10xf32>) -> ()
|
| D | legalize-tf-variables.mlir | 19 // CHECK: "tfl.assign_variable"(%[[RESOURCE]], %[[ADD]]) : (tensor<1xi32>, tensor<1x10xf32>) -> ()
52 … // CHECK: "tfl.assign_variable"(%[[RESOURCE]], %[[ADD]]) : (tensor<1xi32>, tensor<1x10xf32>) -> ()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.td | 4574 def TFL_AssignVariableOp : TFL_Op<"assign_variable", []> {
|