- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 293 for runOnOperation (0.18 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/mark_functions_noinline.cc
"the attribute `tf._noinline = true`. This attributes the function " "from being inlined by the `InlinerPass`."; } void runOnOperation() override; private: ListOption<std::string> CreateNoinlineFunctionsOption( const ArrayRef<std::string> noinline_functions) { return {*this, "noinline-functions", llvm::cl::desc(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 18 02:52:57 UTC 2023 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/strip_saved_module_metadata.cc
#include "tensorflow/compiler/mlir/tensorflow/transforms/tf_savedmodel_passes.h.inc" struct StripSavedModuleMetadataPass : public impl::StripSavedModuleMetadataPassBase< StripSavedModuleMetadataPass> { void runOnOperation() override; }; bool ShouldStripAttr(NamedAttribute &namedAttr) { auto name = namedAttr.getName().strref(); return name.starts_with("tf_saved_model."); } void StripModule(Operation *module) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 08:06:04 UTC 2023 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/strip_tf_attributes.cc
#define GEN_PASS_DEF_STRIPTFATTRIBUTESPASS #include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc" struct StripTfAttributesPass : 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;
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/transforms/order_by_dialect.cc
public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(OrderByDialectPass) void runOnOperation() override; }; int DialectOrdering(Operation* predecessor, Operation* op) { return predecessor && predecessor->getName().getDialectNamespace() == op->getName().getDialectNamespace(); } void OrderByDialectPass::runOnOperation() { ModuleOp module = getOperation();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 08 17:01:11 UTC 2022 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_session_initializer_to_function.cc
struct ConvertSessionInitializerToFunctionPass : public impl::ConvertSessionInitializerToFunctionPassBase< ConvertSessionInitializerToFunctionPass> { void runOnOperation() override; }; void ConvertSessionInitializerToFunctionPass::runOnOperation() { ModuleOp module = getOperation(); auto session_initializer = tf_saved_model::GetSessionInitializerOp(module); if (!session_initializer) return;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 01 05:03:09 UTC 2023 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/testing/test_tf_to_stablehlo_pass.cc
: public impl::TestTFToStablehloPassBase<TestTFToStablehloPass> { public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(TestTFToStablehloPass) private: void runOnOperation() override; }; void TestTFToStablehloPass::runOnOperation() { ModuleOp module_op = getOperation(); MLIRContext* ctx = &getContext(); mlir::PassManager pm(ctx); AddTFToStablehloPasses(pm); if (!RunPassesOnModuleOp(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 17:19:36 UTC 2024 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_gpu.cc
return "Suitable transformation for gpu only."; } void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect>(); } void runOnOperation() override; }; void DeviceTransformGPUPass::runOnOperation() { auto func = getOperation(); auto* ctx = &getContext(); RewritePatternSet patterns = GetHardwareRewritePatternsGPU(ctx);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 07 18:43:51 UTC 2022 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_rewrite.cc
namespace { #define GEN_PASS_DEF_XLAREWRITEPASS #include "tensorflow/compiler/mlir/tensorflow/transforms/tf_device_passes.h.inc" struct XlaRewritePass : public impl::XlaRewritePassBase<XlaRewritePass> { void runOnOperation() override; }; void MoveResourceArgsToEnd(func::FuncOp callee) { llvm::DenseMap<BlockArgument, BlockArgument> mapping; unsigned num_params = callee.getNumArguments(); llvm::BitVector removed_params(num_params);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/device_index_selector.cc
// for CPU or GPU. struct DeviceIndexSelector : public impl::DeviceIndexSelectorPassBase<DeviceIndexSelector> { void runOnOperation() override; }; } // namespace void DeviceIndexSelector::runOnOperation() { func::FuncOp func = getOperation(); // Convert all the DeviceIndex ops to constant values. func.getBody().walk([](TF::DeviceIndexOp op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/raise_custom_ops.cc
explicit RaiseCustomOpsPass() {} explicit RaiseCustomOpsPass(const std::vector<std::string> &target_ops) { this->target_ops_ = target_ops; } void runOnOperation() override; }; void RaiseCustomOpsPass::runOnOperation() { auto fn = getOperation(); OpBuilder builder(fn.getContext()); absl::flat_hash_set<std::string> target_op_names(target_ops_.begin(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 17 07:31:01 UTC 2023 - 4.3K bytes - Viewed (0)