Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 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/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)
  4. 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)
  5. 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)
  6. 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)
  7. 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