Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 14 for getSharedName (0.89 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/quantization/tensorflow/passes/merge_duplicate_resource_ops.cc

                    TF::MutableHashTableV2Op>(resource_op)) {
        return resource_op;
      }
      return nullptr;
    }
    
    // Returns the `shared_name` attribute value if exists. If not, returns an
    // empty string.
    StringRef GetSharedName(Operation* op) {
      if (!op->hasAttrOfType<StringAttr>(kSharedNameAttr)) return "";
      return op->getAttrOfType<StringAttr>(kSharedNameAttr).getValue();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 26 04:26:16 UTC 2023
    - 4.7K bytes
    - Viewed (0)
  5. 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)
  6. 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)
  7. tensorflow/compiler/mlir/tensorflow/transforms/resource_analyzer_test_pass.cc

      if (auto attr = var_handle_op->getAttrOfType<StringAttr>("device")) {
        resource_key.device = attr.getValue();
      }
    
      resource_key.container = var_handle_op.getContainer();
      resource_key.shared_name = var_handle_op.getSharedName();
    
      return resource_key;
    }
    
    // Prints the analysis result for each resource ops found in `module_op` for
    // testing purposes.
    void PrintAnalysisResults(const TF::ResourceAnalyzer& analyzer,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 03 00:21:29 UTC 2023
    - 3K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_restore_op.cc

      SmallVector<Type> tensor_types{};
      SmallVector<std::string> tensor_names{};
      for (auto var_handle_op : target_var_handle_ops) {
        tensor_names.emplace_back(var_handle_op.getSharedName().str());
        // Location must be set to the same name as the shared name. The Location is
        // later tranlated to the op's name when exported to `GraphDef`. This is
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun Mar 12 06:02:20 UTC 2023
    - 9K bytes
    - Viewed (0)
  9. 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)
  10. 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)
Back to top