Searched defs:var_handle_op (Results 1 – 10 of 10) sorted by relevance
28 std::string GetVariableName(TF::VarHandleOp var_handle_op) { in GetVariableName()41 for (auto var_handle_op : var_handle_ops) in GetVariableNames() local46 tensorflow::Var* GetVariableFromSession(mlir::TF::VarHandleOp var_handle_op, in GetVariableFromSession()
66 TF::VarHandleOp var_handle_op, const tensorflow::Tensor& resource_tensor, in GetTensorValueAsElementsAttr()158 TF::VarHandleOp var_handle_op, ElementsAttr value, in ReplaceVarWithConstant()225 for (auto var_handle_op : var_handle_ops) in RemoveVariablesInitializations() local229 for (auto var_handle_op : sesion_init_func.getOps<TF::VarHandleOp>()) { in RemoveVariablesInitializations() local324 for (auto var_handle_op : func.getOps<TF::VarHandleOp>()) { in FreezeVariables() local355 auto var_handle_op = std::get<0>(variable_value_pair); in FreezeVariables() local420 for (auto var_handle_op : variables) { in FreezeVariables() local
39 bool IsVariableInitialized(mlir::TF::VarHandleOp var_handle_op, in IsVariableInitialized()64 for (auto var_handle_op : block.getOps<TF::VarHandleOp>()) in MarkInitializedVariablesInFunction() local
38 void InitializeVariable(TF::VarHandleOp var_handle_op, in InitializeVariable()118 for (auto var_handle_op : func_op.getOps<TF::VarHandleOp>()) { in InitializeVariablesInSessionInitializer() local
76 mlir::LogicalResult ValidateVarHandle(TF::VarHandleOp var_handle_op) { in ValidateVarHandle()111 bool VariableIsInitialized(TF::VarHandleOp var_handle_op) { in VariableIsInitialized()132 for (auto var_handle_op : in PromoteVarHandlesToArguments() local
161 VarHandleOp var_handle_op = builder.create<VarHandleOp>( in runOnOperation() local
2110 if (auto var_handle_op = dyn_cast<VarHandleOp>(op)) { in InferShapeForSingleOperation() local
174 Node* var_handle_op = nullptr; in Run() local
188 if (auto var_handle_op = llvm::dyn_cast_or_null<mlir::TF::VarHandleOp>( in CanHoist() local591 auto var_handle_op = builder.create<mlir::TF::VarHandleOp>( in ConvertReferenceVariableToResourceVariable() local
692 Node** var_handle_op = nullptr) { in MakeRead()