Home
last modified time | relevance | path

Searched defs:assign_variable (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dtpu_add_resource_device_attribute.cc77 auto assign_variable = llvm::dyn_cast<mlir::TF::AssignVariableOp>(user); in IdentifyConnectedAssignVariableOp() local
113 mlir::Operation* assign_variable = in runOnOperation() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dassign_variable.cc27 namespace assign_variable { namespace
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/
H A Dresource_value_typed_analyzer.cc118 if (auto assign_variable = dyn_cast<TF::AssignVariableOp>(op)) { in AnalyzeRegion() local