- Sort Score
- Result 10 results
- Languages All
Results 81 - 90 of 240 for StringRef (0.19 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/insert_calibration_statistics_saver.cc
void FindCustomAggregatorOps( Region& region, const std::unordered_set<std::string>& aggregator_ops_to_ignore, SmallVector<Value>& statistics_outputs, SmallVector<StringRef>& ids, SmallVector<int32_t>& calibration_methods) { for (auto op : region.getOps<TF::CustomAggregatorOp>()) { if (aggregator_ops_to_ignore.count(op.getId().str())) continue; ids.push_back(op.getId());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_dialect_to_executor.cc
#include "tensorflow/compiler/mlir/tf2xla/api/v2/tf_dialect_to_executor.h" #include <memory> #include <string> #include "absl/log/log.h" #include "absl/status/status.h" #include "llvm/ADT/StringRef.h" #include "mlir/Dialect/Func/IR/FuncOps.h" // from @llvm-project #include "mlir/IR/BuiltinOps.h" // from @llvm-project #include "mlir/Pass/PassManager.h" // from @llvm-project
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 13 23:22:50 UTC 2024 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_mlir.cc
#include "tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_mlir.h" #include <memory> #include <string> #include <vector> #include "absl/log/log.h" #include "llvm/ADT/StringRef.h" #include "mlir/IR/BuiltinOps.h" // from @llvm-project #include "mlir/IR/DialectRegistry.h" // from @llvm-project #include "mlir/IR/MLIRContext.h" // from @llvm-project
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 20:29:34 UTC 2024 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_restore_op.cc
explicit InsertRestoreOpPass() = default; // The argument used to refer to the pass in the textual format (e.g. on the // commandline). StringRef getArgument() const final { return "quant-insert-restore-op"; } StringRef getDescription() const final { return "Creates RestoreV2 op to initialize the variables in the " "initializer function (`tf_saved_model.initializer_type == "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Mar 12 06:02:20 UTC 2023 - 9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_dialect.h
// Dialect is used to model TF GradientDef. GetGradientAttrName() returns the // string description of gradient attribute. static StringRef GetGradientAttrName() { return "tf.gradient"; } // This attribute marks if a function is stateful. // Returns the string description of stateful attribute. static StringRef GetStatefulAttrName() { return "tf.signature.is_stateful"; } // Returns true if the op can be duplicated during transformations.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 01:19:25 UTC 2023 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/device_target.cc
} } LogicalResult DeviceTarget::RegisterKernel( llvm::StringRef kernel, const KernelSpecs::Signature& signature, const ScaleFn& fn, const ScaleDecomposeFn& dfn) { return specs_[kernel].Add(signature, {ScaleConstraintType::CustomScale, fn}); } namespace ph = std::placeholders; LogicalResult DeviceTarget::RegisterKernel( llvm::StringRef kernel, const KernelSpecs::Signature& signature,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 08 10:41:08 UTC 2024 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_tfl_passes.h
// variables. If the `saved_model_dir` directory path is provided, then the // `tf_saved_model.asset` ops will be freezed. void AddTFToTFLConversionPasses(llvm::StringRef saved_model_dir, const toco::TocoFlags& toco_flags, const mlir::TFL::PassConfig& pass_config, mlir::OpPassManager* pass_manager);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 01 06:14:07 UTC 2024 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/cast_bf16_ops_to_f32.cc
MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(CastBf16OpsToF32Pass) explicit CastBf16OpsToF32Pass() = default; StringRef getArgument() const final { // This is the argument used to refer to the pass in // the textual format (on the commandline for example). return "quant-cast-bf16-ops-to-f32"; } StringRef getDescription() const final { return "Cast BF16 operations to F32."; } void runOnOperation() override;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Dec 10 05:52:02 UTC 2023 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_with_tf2xla_passes.h
#define TENSORFLOW_COMPILER_MLIR_TF2XLA_TRANSFORMS_LEGALIZE_TF_WITH_TF2XLA_PASSES_H_ #include <memory> #include <optional> #include "llvm/ADT/StringRef.h" #include "mlir/IR/MLIRContext.h" // from @llvm-project #include "mlir/IR/PatternMatch.h" // from @llvm-project #include "mlir/Pass/Pass.h" // from @llvm-project #include "mlir/Pass/PassRegistry.h" // from @llvm-project
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/tf_stablehlo_pass.cc
mlir::stablehlo::registerAllDialects(registry); registry.insert<mlir::func::FuncDialect, mlir::arith::ArithDialect>(); registry.insert<shape::ShapeDialect>(); } public: StringRef getArgument() const final { return "tf-mhlo"; } StringRef getDescription() const final { return "This pass will legalize TF Ops to MHLO Ops."; } protected: Option<bool> skip_quantization_ops_{ *this, "skip-quantization-ops",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 7.5K bytes - Viewed (0)