Searched defs:read_variable_op (Results 1 – 3 of 3) sorted by relevance
183 ReadVariableOp read_variable_op = builder.create<ReadVariableOp>( in runOnFunction() local
84 void PropagateUsage(TF::ReadVariableOp read_variable_op, ElementsAttr value) { in PropagateUsage()103 if (auto read_variable_op = dyn_cast<TF::ReadVariableOp>(user_op)) { in PropagateUsage() local
124 if (auto read_variable_op = dyn_cast<TF::ReadVariableOp>(&op)) { in ForwardStoreToLoad() local