Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 9 of 9 for getSharedName (0.19 sec)

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

      StringRef shared_name = GetSharedName(hash_table);
      if (shared_name.empty()) return false;
    
      for (func::FuncOp init_func_op :
           tf_saved_model::GetInitializerFunctions(module_op)) {
        for (Operation& op : init_func_op.getBody().getOps()) {
          StringRef other_shared_name = GetSharedName(&op);
          if (IsHashTableOp(&op) && other_shared_name == shared_name) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 8.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/utils/session_utils.cc

      // then fallback to shared_name attribute.
      if (auto loc = mlir::dyn_cast<NameLoc>(var_handle_op->getLoc()))
        return loc.getName().str();
      return var_handle_op.getSharedName().str();
    }
    
    std::vector<std::string> GetVariableNames(
        llvm::ArrayRef<TF::VarHandleOp> var_handle_ops) {
      std::vector<std::string> names;
      names.reserve(var_handle_ops.size());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 3.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/tensorflow/cc/save_variables.cc

          !status.ok()) {
        return status;
      }
    
      if (!bundle_writer.Add(/*key=*/var_handle_op.getSharedName(), const_tensor)
               .ok()) {
        return bundle_writer.status();
      }
    
      return var_handle_op.getSharedName().str();
    }
    
    }  // namespace
    
    absl::StatusOr<std::vector<std::string>> SaveVariablesToCheckpoint(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Feb 26 03:36:55 UTC 2024
    - 4.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/localize_var_handles.cc

      } else if (auto handle = llvm::dyn_cast<TF::VarHandleOp>(source)) {
        container = handle.getContainer();
        shared_name = handle.getSharedName();
      } else if (auto it = llvm::dyn_cast<TF::IteratorOp>(source)) {
        container = it.getContainer();
        shared_name = it.getSharedName();
      } else {
        // Can't happen, as long as this file and resource_dataflow.cc are in sync.
        return;
      }
    
      OpBuilder builder(op);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 06 23:53:00 UTC 2024
    - 5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/transforms/quantize_variables.cc

            {mlir::cast<TensorType>(ref_qtype)}, builder.getContext()));
        auto new_var_handle_op = builder.create<VarHandleOp>(
            var_handle_op.getLoc(), output_type, var_handle_op.getContainer(),
            var_handle_op.getSharedName());
        var_handle_op->replaceAllUsesWith(new_var_handle_op);
        var_handle_op.erase();
      }
    }
    
    void QuantizeVariablesPass::runOnOperation() {
      ResourceIdMap resource_id_map;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.5K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/ir/host_runtime/tfrt_ops.cc

          resource_vec.push_back(GetResourceHandleValueAndIdBase(
              mlir::cast<mlir::StringAttr>(getContainer()[index]).getValue(),
              mlir::cast<mlir::StringAttr>(getSharedName()[index]).getValue(),
              device, getResults()[index], resource_handle_id_map, next_id));
        }
      }
      return resource_vec;
    }
    
    LogicalResult _TfrtGetResourceOp::verify() {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.6K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/transforms/legalize_hashtables.cc

        // native resource design is based on integer keys to identify the
        // corresponding resource objects.
        auto table_id =
            static_cast<int32_t>(::llvm::hash_value(hashtable_op.getSharedName()));
        auto key_dtype = hashtable_op.getKeyDtype();
        auto value_dtype = hashtable_op.getValueDtype();
    
        rewriter.replaceOpWithNewOp<TFL::HashtableOp>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7.6K bytes
    - Viewed (0)
  8. 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)
  9. 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)
Back to top