- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 217 for StringRef (0.16 sec)
-
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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)