Searched refs:assign_variable (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | assign_variable.cc | 27 namespace assign_variable { namespace
69 static TfLiteRegistration r = {nullptr, nullptr, assign_variable::Prepare, in Register_ASSIGN_VARIABLE()
70 assign_variable::Eval}; in Register_ASSIGN_VARIABLE()
|
| D | BUILD | 550 "assign_variable.cc",
716 "assign_variable.cc",
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/ |
| D | side_effects.mlir | 3 // CHECK-LABEL: func @assign_variable
5 func @assign_variable() {
|
| /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/analysis/ |
| D | resource_value_typed_analyzer.cc | 117 if (auto assign_variable = dyn_cast<TF::AssignVariableOp>(op)) { in AnalyzeRegion() local
118 SetPotentiallyWritten(assign_variable.resource()); in AnalyzeRegion()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | legalize-tf-variables.mlir | 22 …// CHECK: "tfl.assign_variable"(%[[RESOURCE]], %[[ADD]]) : (tensor<!tf_type.resource<tensor<1x10xf…
45 …// CHECK: "tfl.assign_variable"(%[[RESOURCE]], %[[CST]]) : (tensor<!tf_type.resource<tensor<1x10xf…
63 …// CHECK: "tfl.assign_variable"(%[[RESOURCE]], %[[ADD]]) : (tensor<!tf_type.resource<tensor<1x10xf…
|
| /external/tensorflow/tensorflow/lite/ |
| D | Android.bp | 77 "kernels/assign_variable.cc",
|
| D | tflite_static.bp | 49 "kernels/assign_variable.cc",
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.td | 4781 def TFL_AssignVariableOp : TFL_Op<"assign_variable", []> {
|