Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 217 for StringRef (0.16 sec)

  1. tensorflow/compiler/mlir/op_or_arg_name_mapper.cc

    static inline absl::string_view StringRefToView(llvm::StringRef ref) {
      return absl::string_view(ref.data(), ref.size());
    }
    
    static inline llvm::StringRef StringViewToRef(absl::string_view view) {
      return llvm::StringRef(view.data(), view.size());
    }
    
    namespace tensorflow {
    
    OpOrArgNameMapper::~OpOrArgNameMapper() = default;
    
    llvm::StringRef OpOrArgNameMapper::GetUniqueName(llvm::StringRef prefix,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args.cc

    // SavedModel loader attaches the prefix for you during loading.
    StringRef MaybeStripAssetDirectoryPrefix(const StringRef filename) {
      if (filename.find("assets/") == 0) {
        return filename.drop_front(7);
      } else {
        return filename;
      }
    }
    
    AssetFileDef CreateAssetFileDef(const StringRef filename,
                                    const StringRef tensor_name) {
      AssetFileDef asset_file_def{};
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_hashtable_ops_as_args.cc

    namespace mlir {
    namespace quant {
    namespace {
    
    constexpr StringRef kSharedNameAttr = "shared_name";
    
    class LiftHashTableOpsAsArgsPass
        : public PassWrapper<LiftHashTableOpsAsArgsPass, OperationPass<ModuleOp>> {
     public:
      MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(LiftHashTableOpsAsArgsPass)
      explicit LiftHashTableOpsAsArgsPass() = default;
    
      StringRef getArgument() const final {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 8.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/tpu_device_propagation.cc

    // devices, and empty StringRef will be returned. Control dependencies,
    // NextIteration.Source -> NextIteration.Sink token dependencies, and
    // LoopCond -> Switch data dependencies are ignored.
    llvm::StringRef FindDeviceFromOperands(
        Operation& op,
        const llvm::DenseMap<Value, llvm::StringRef>& value_to_device) {
      llvm::StringRef new_device;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/readonly_references_to_resources.cc

    #include "tensorflow/compiler/mlir/tensorflow/transforms/passes.h"
    
    namespace mlir {
    namespace TF {
    namespace {
    
    // Location attribute.
    constexpr StringRef kClassAttr = "_class";
    constexpr StringRef kSharedNameAttr = "shared_name";
    constexpr StringRef kLocationPrefix = "loc:@";
    
    #define GEN_PASS_DEF_CONVERTREADONLYREFERENCEVARIABLESTORESOURCEVARIABLESPASS
    #include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7.8K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.h

        mlir::ModuleOp module_op, llvm::StringRef device_type,
        xla::XlaComputation* xla_computation, bool use_tuple_args,
        bool enable_op_fallback, bool return_tuple,
        const XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns =
            {},
        llvm::MutableArrayRef<std::unique_ptr<mlir::Pass>>
            custom_legalization_passes = {},
        llvm::StringRef module_name = llvm::StringRef());
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 17:24:39 UTC 2024
    - 10.4K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_island_coarsening.cc

    bool is_valid_special_tpu_op(
        std::vector<IslandOp>& ops, llvm::StringRef cluster_name,
        llvm::SmallDenseMap<llvm::StringRef, llvm::SmallDenseSet<Operation*>>&
            cluster_to_tpu_op_map) {
      for (IslandOp op : ops) {
        Operation* wrapped_op = &op.GetBody().front();
        std::optional<llvm::StringRef> wrapped_op_cluster_name =
            GetTpuClusterName(wrapped_op);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 27.6K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/quantization/tensorflow/passes/add_quantization_unit_loc.cc

        // Op type is a NameLoc with the ":" suffix.
        StringRef op_type_with_suffix =
            mlir::cast<NameLoc>(locations.front()).getName().strref();
        StringRef op_type =
            op_type_with_suffix.substr(0, op_type_with_suffix.size() - 1);
        new_unit.set_op_type(op_type.str());
    
        if (isa<NameLoc>(locations.back())) {
          StringRef name_loc_id =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7.4K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/utils/attribute_utils.h

    // associated compilation cluster and replication metadata op.
    inline constexpr llvm::StringRef kTpuReplicateAttr = "_tpu_replicate";
    // Device types.
    inline constexpr llvm::StringRef kTpuDevice = "TPU";
    // _xla_outside_compilation
    inline constexpr llvm::StringRef kXlaOutsideCompilationAttr =
        "_xla_outside_compilation";
    // device attr
    inline constexpr llvm::StringRef kDeviceAttr = "device";
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 22 19:47:48 UTC 2024
    - 8.7K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc

      void CountOp(DenseMap<StringRef, llvm::StringMap<int64_t>> &op_count_map,
                   StringRef op_name, StringRef dtype);
    
     private:
      llvm::StringMap<int64_t> op_with_dialect_count_;
      DenseMap<StringRef, llvm::StringMap<int64_t>> op_dtype_count_;
      llvm::StringMap<int64_t> dialect_count_;
      llvm::StringMap<StringRef> dialect_name_of_;
      llvm::StringMap<StringRef> op_name_of_;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.7K bytes
    - Viewed (0)
Back to top