- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for globalBuilder (0.44 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/lower_variable_ops_to_ml_program.cc
SymbolTable symbol_table(module); OpBuilder globalBuilder(module.getBodyRegion()); module.walk([&](TF::ReadVariableOp op) { Operation* source = GetHandleSource(op, solver); if (!source) return; ml_program::GlobalOp globalOp = CreateGlobalOpFromOp(source, globalBuilder, symbol_table); if (!globalOp) return; OpBuilder builder(op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 06 23:53:00 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_globals_to_ml_program.cc
} else { initial_value = mlir::Attribute(); } opToName[globalTensor] = name; auto variableOp = globalBuilder.create<ml_program::GlobalOp>( globalTensor.getLoc(), name, globalTensor.getType(), globalTensor.getIsMutable(), initial_value, /*visibility=*/globalBuilder.getStringAttr("private")); variableOp.setPrivate(); } SymbolTable syms(module);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/localize_var_handles.cc
solver.load<dataflow::SparseConstantPropagation>(); TF::LoadResourceDataflowAnalysis(solver); if (failed(solver.initializeAndRun(module))) return signalPassFailure(); OpBuilder globalBuilder(module.getBodyRegion()); module.walk( [&](TF::ReadVariableOp op) { MaybeCreateVarHandleForOp(op, solver); }); module.walk( [&](TF::AssignVariableOp op) { MaybeCreateVarHandleForOp(op, solver); });
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 06 23:53:00 UTC 2024 - 5K bytes - Viewed (0)