Searched refs:assign_variable (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | tpu_add_resource_device_attribute.cc | 77 auto assign_variable = llvm::dyn_cast<mlir::TF::AssignVariableOp>(user); in IdentifyConnectedAssignVariableOp() local
78 if (!assign_variable) continue; in IdentifyConnectedAssignVariableOp()
80 return assign_variable; in IdentifyConnectedAssignVariableOp()
113 mlir::Operation* assign_variable = in runOnOperation() local
115 if (assign_variable == nullptr) continue; in runOnOperation()
118 llvm::cast<mlir::TF::AssignVariableOp>(assign_variable) in runOnOperation()
|
| /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 | 676 "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.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 | 118 if (auto assign_variable = dyn_cast<TF::AssignVariableOp>(op)) { in AnalyzeRegion() local
119 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 | 51 "kernels/assign_variable.cc",
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.td | 5241 def TFL_AssignVariableOp : TFL_Op<"assign_variable", []> {
|