Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dresource_op_lifting.cc221 if (auto assign_variable_op = dyn_cast<TF::AssignVariableOp>(&op)) { in ForwardStoreToLoad() local
507 for (auto assign_variable_op : llvm::make_early_inc_range(assign_ops)) { in AppendResourceStoreValueToReturn() local
836 for (auto assign_variable_op : llvm::make_early_inc_range(assign_ops)) { in LiftArgRetResourcesForFunction() local