- Sort Score
- Result 10 results
- Languages All
Results 101 - 110 of 226 for StringRef (0.13 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tf_device_assignment.cc
SimpleTFDeviceAssignmentPass> { public: SimpleTFDeviceAssignmentPass() = default; SimpleTFDeviceAssignmentPass(const SimpleTFDeviceAssignmentPass&) {} explicit SimpleTFDeviceAssignmentPass(llvm::StringRef default_device) { default_device_ = std::string(default_device); } void runOnOperation() override { Builder builder(&getContext()); Dialect* tf = getContext().getLoadedDialect<TensorFlowDialect>();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/hoist_loop_invariant.cc
ResourceHandle GetResourceHandle(Operation *op) { llvm::StringRef device; if (auto attr = op->getAttrOfType<StringAttr>("device")) { device = attr.getValue(); } llvm::StringRef container; if (auto attr = op->getAttrOfType<StringAttr>("container")) { container = attr.getValue(); } llvm::StringRef shared_name; if (auto attr = op->getAttrOfType<StringAttr>("shared_name")) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/utils/utils.h
// tf.ConcatV2 => tf__concat_v2 // TODO(fengliuai): move to an util file. std::string GetComposeFuncName(StringRef tf_op_name); // This is a hardcoded rule for mapping a TFR function op name to the // corresponding TF opname. Examples: // tf__pack -> tf.Pack // tf__concat_v2 => tf.ConcatV2 std::string GetTFOpName(StringRef compose_func_name); // Validate the attributes of 'src' is either contained in the registered
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 03:08:33 UTC 2023 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_reorder_replicate_and_partitioned_inputs.cc
limitations under the License. ==============================================================================*/ #include <cstddef> #include <optional> #include "llvm/ADT/STLExtras.h" #include "llvm/ADT/StringRef.h" #include "mlir/Pass/Pass.h" // from @llvm-project #include "mlir/Support/LogicalResult.h" // from @llvm-project #include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 24 23:08:55 UTC 2023 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_serialization.cc
// `tf.backend_config` is a DictionaryAttr, JAX2TF sets the value of its // i64 attribute `called_index` to the TF function's name. constexpr llvm::StringRef kTfBackendConfigAttrName = "tf.backend_config"; constexpr llvm::StringRef kCalledIndexAttrName = "called_index"; constexpr llvm::StringRef kCalledFuncAttrName = "called_func"; // Converts `called_func` attributes in custom call ops back to `called_index`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/tac_filter.cc
} explicit TacFilterPass(TacFilters* tac_filters) { tac_filters_ = tac_filters; } private: TacFilters* tac_filters_ = nullptr; llvm::StringRef getArgument() const final { return "tfl-tac-filter"; } llvm::StringRef getDescription() const final { return "This pass marks the ops to skip target annotation by inserting " "`tac.skip_target_annotation` attribute to them based on user "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/strip_tf_attributes.cc
: public impl::StripTfAttributesPassBase<StripTfAttributesPass> { void runOnOperation() override; }; bool ShouldStripAttr(NamedAttribute &namedAttr) { StringRef name = namedAttr.getName().strref(); if (name.starts_with("tf.") || name.starts_with("tf_")) return true; StringRef value = namedAttr.getValue().getDialect().getNamespace(); return value == "tf" || value.starts_with("tf_"); } void StripFunction(func::FuncOp func) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 08:06:04 UTC 2023 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/session_utils.cc
#include "tensorflow/compiler/mlir/tensorflow/utils/session_utils.h" #include <string> #include <vector> #include "absl/status/status.h" #include "llvm/ADT/SmallSet.h" #include "llvm/ADT/StringRef.h" #include "mlir/Support/LLVM.h" // from @llvm-project #include "tensorflow/compiler/mlir/utils/string_container_utils.h" #include "tensorflow/core/common_runtime/device_mgr.h" #include "tensorflow/core/framework/device.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_types.def
HANDLE_TF_REF_TYPE(Int32Ref, INT32_REF, "int32ref") HANDLE_TF_REF_TYPE(Uint64Ref, UINT64_REF, "uint64ref") HANDLE_TF_REF_TYPE(Int64Ref, INT64_REF, "int64ref") HANDLE_TF_REF_TYPE(StringRef, STRING_REF, "stringref") HANDLE_TF_REF_TYPE(BoolRef, BOOL_REF, "boolref") HANDLE_TF_REF_TYPE(Quint8Ref, QUINT8_REF, "quint8ref") HANDLE_TF_REF_TYPE(Qint8Ref, QINT8_REF, "qint8ref") HANDLE_TF_REF_TYPE(Quint16Ref, QUINT16_REF, "quint16ref")
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Sep 09 03:12:53 UTC 2023 - 3.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_utils.cc
#include <algorithm> #include <array> #include <cstdint> #include <optional> #include <string> #include <utility> #include <vector> #include "llvm/ADT/StringRef.h" #include "mlir/IR/Builders.h" // from @llvm-project #include "mlir/IR/BuiltinAttributes.h" // from @llvm-project #include "mlir/IR/BuiltinTypeInterfaces.h" // from @llvm-project
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 18:33:05 UTC 2024 - 3.4K bytes - Viewed (0)