Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for globalBuilder (0.51 sec)

  1. 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)
  2. 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)
  3. 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)
Back to top