Searched defs:var_handle_op (Results 1 – 4 of 4) sorted by relevance
108 mlir::LogicalResult ValidateVarHandle(TF::VarHandleOp var_handle_op) { in ValidateVarHandle()155 for (auto var_handle_op : in PromoteVarHandlesToArguments() local
148 VarHandleOp var_handle_op = builder.create<VarHandleOp>( in runOnFunction() local
178 Node* var_handle_op = nullptr; in Run() local
653 Node** var_handle_op = nullptr) { in MakeRead()