Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 14 of 14 for getSharedName (0.16 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/lower_variable_ops_to_ml_program.cc

    namespace {
    
    std::string GetVariableName(Operation* op) {
      if (auto handle = dyn_cast<TF::VarHandleOp>(op)) {
        std::string container = handle.getContainer().str();
        std::string shared_name = handle.getSharedName().str();
        if (container.empty()) {
          return absl::StrCat("vars.", shared_name);
        } else {
          return absl::StrCat("vars.", container, ".", shared_name);
        }
    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/quantization/tensorflow/passes/insert_save_op.cc

      SmallVector<std::string> tensor_names = {};
      SmallVector<Value> tensor_values = {};
      for (auto var_handle_op : var_handle_ops) {
        tensor_names.emplace_back(var_handle_op.getSharedName().str());
    
        auto read_var_op = builder.create<TF::ReadVariableOp>(
            var_handle_op.getLoc(), var_handle_op.resource_subtype(),
            var_handle_op);
        tensor_values.emplace_back(read_var_op.getResult());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 22 05:52:39 UTC 2024
    - 9.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting_cleanup.cc

      }
      if (TF::VarHandleOp var_handle_op = llvm::dyn_cast<TF::VarHandleOp>(&op)) {
        auto ANONYMOUS_NAME = ::tensorflow::ResourceHandle::ANONYMOUS_NAME;
        if (var_handle_op.getSharedName() == ANONYMOUS_NAME) {
          return std::make_optional(LocalVarOp(var_handle_op));
        }
      }
      return {};
    }
    
    // Eliminate local variables that are only assigned to but never read, and thus
    // are dead.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

        llvm::SmallDenseMap<ResourceHandle, int64_t> &resource_handle_id_map,
        int64_t &next_id) {
      llvm::StringRef device = GetDeviceOrEmpty(getOperation());
      return {GetResourceHandleValueAndIdBase(getContainer(), getSharedName(),
                                              device, getWriter(),
                                              resource_handle_id_map, next_id)};
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
Back to top