Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for setPrivate (0.13 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_save_op.cc

      auto save_func = builder.create<func::FuncOp>(
          NameLoc::get(builder.getStringAttr(kTfQuantSaveFuncName)),
          /*sym_name=*/kTfQuantSaveFuncName, func_type);
      save_func.addEntryBlock();
      save_func.setPrivate();
    
      return save_func;
    }
    
    // Creates a save function that contains the `TF::SaveV2Op` for the variables in
    // `var_handle_ops`. The `var_handle_ops` are cloned into the new function and
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 22 05:52:39 UTC 2024
    - 9.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/lower_globals_to_ml_program.cc

            globalTensor.getLoc(), name, globalTensor.getType(),
            globalTensor.getIsMutable(), initial_value,
            /*visibility=*/globalBuilder.getStringAttr("private"));
        variableOp.setPrivate();
      }
    
      SymbolTable syms(module);
      for (auto func : module.getOps<func::FuncOp>()) {
        if (!tf_saved_model::IsExported(func)) {
          continue;
        }
        bool success = true;
    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/cluster_outlining.cc

          func::FuncOp::create(op.getLoc(), func_name, func_type);
    
      // This function is not externally visible and marking it private would allow
      // symbol-dce pass to remove it when it is not referenced anymore.
      outlined_func.setPrivate();
    
      // Create function body.
      Block* outlined_func_block = outlined_func.addEntryBlock();
    
      // Replace uses of live-in values within cluster_op region with function
      // arguments.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 7.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_quantized_functions.cc

        if (symbol_table.lookup(func.getSymName()) != nullptr) continue;
    
        // Set the function to private and insert to the module.
        func::FuncOp new_func = func.clone();
        new_func.setPrivate();
        symbol_table.insert(new_func);
    
        // For consistency, we require all quantized composite function to have
        // the "tf_quant.quantized_ops" attribute.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 22 05:52:39 UTC 2024
    - 8.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_serialization.cc

              // Entry function must be public and has symbol name "@main".
              cloned.setPublic();
              cloned.setName(kStablehloMainFunctionName);
            } else {
              cloned.setPrivate();
            }
    
            return WalkResult::advance();
          },
          &symbol_table);
      if (result.wasInterrupted()) {
        return failure();
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 9.5K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/stablehlo/odml_converter/transforms/outline_composites.cc

      auto name = rewriter.getStringAttr("gelu_decomp");
      func::FuncOp new_func = rewriter.create<func::FuncOp>(
          insertion_point->front().getLoc(), name, ftype);
      new_func.setPrivate();
      new_func.addEntryBlock();
      rewriter.setInsertionPointToStart(&new_func.getBody().front());
    
      auto one_val = DenseElementsAttr::get(type, kOne);
      auto one_cst =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 9.6K bytes
    - Viewed (0)
Back to top