Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 29 for Fremovexattr (0.19 sec)

  1. tensorflow/compiler/mlir/lite/experimental/tac/transforms/tac_filter.cc

        if (type == FunctionFilter::SKIP_TARGET_ANNOTATION) {
          op.setAttr(kSkipTargetAnnotation, builder.getUnitAttr());
        } else if (type == FunctionFilter::INCLUDE_TARGET_ANNOTATION) {
          op.removeAttr(kSkipTargetAnnotation);
        }
      }
    }
    
    void ApplyTacFilter(ModuleOp module, const TacFilter& tac_filter,
                        SmallVector<Operation*>& filtered_ops, OpBuilder& builder) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc

      // nested ops.
      for (Operation* cluster_op : cluster_ops) {
        cluster_op->moveBefore(body, body->end());
        cluster_op->walk([&](Operation* inner_op) {
          inner_op->removeAttr(mlir::TF::kReplicationInfoAttr);
          inner_op->removeAttr(mlir::TF::kCompileDeviceTypeAttr);
    
          if (auto attr = inner_op->getAttrOfType<StringAttr>(kDeviceAttr)) {
            // Preserve device attribute if the op is placed on a replicated core
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 22:03:30 UTC 2024
    - 39.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/cluster_outlining.cc

      std::string func_name;
      if (auto outlined_func_name = op->template getAttrOfType<StringAttr>(
              TF::kClusterOutlinedFunctionNameAttr)) {
        op->removeAttr(TF::kClusterOutlinedFunctionNameAttr);
        func_name = outlined_func_name.str();
      } else {
        func_name = "_func";
      }
    
      func::FuncOp outlined_func =
          func::FuncOp::create(op.getLoc(), func_name, func_type);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 7.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tf2xla/internal/passes/extract_head_tail_outside_compilation.cc

                               llvm::StringRef host_device) {
      Block* launch_block = new Block;
      for (Operation* head_outside_compiled_op : head_outside_compiled_ops) {
        head_outside_compiled_op->removeAttr(kXlaOutsideCompilationAttr);
        head_outside_compiled_op->moveBefore(launch_block, launch_block->end());
      }
    
      mlir::tf_device::LaunchOp launch = CreateLaunchForBlock(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 20.6K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/stablehlo/passes/lift_quantizable_spots_as_functions.cc

              << "quant-stablehlo-lift-quantizable-spots-as-functions failed.";
          signalPassFailure();
        }
      }
    
      // Remove all attr_map attributes.
      module_op.walk([](Operation* op) { op->removeAttr(kAttrMapAttribute); });
    
      // Perform selective quantization. Iterates over the quantization specs and
      // applies quantization methods to each matched lifted function.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 10 04:07:09 UTC 2024
    - 9.3K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_outline_tpu_island.cc

          }
        });
      }
      // Remove `kSkipIslandOutlining` attributes.
      for (func::FuncOp func_op : getOperation().getOps<func::FuncOp>()) {
        if (func_op->hasAttr(mlir::TF::kSkipIslandOutlining)) {
          func_op->removeAttr(mlir::TF::kSkipIslandOutlining);
        }
      }
    }
    
    }  // namespace
    
    std::unique_ptr<OperationPass<ModuleOp>>
    CreateTFExecutorTPUV1IslandOutliningPass() {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.7K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.cc

        Operation* op, PatternRewriter& rewriter) const {
      // removeAttr will return nullptr if the attribute did not exist. Thus we can
      // return success(result) to indicate if this op has changed.
      return success(/*isSuccess=*/
                     op->removeAttr(kDebugModeOpQuantAttrName) ||
                     op->removeAttr(kDebugModeOpFloatAttrName));
    }
    
    }  // namespace quant
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 02:10:16 UTC 2024
    - 43.2K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/quantization/tensorflow/fallback_to_flex_ops.cc

      func.walk([&](Operation *op) {
        if (op->getDialect() != tf_dialect) return;
        if (IsAllowListedOp(op)) return;
        if (op->hasAttr(kNoFallbackAttr)) {
          op->removeAttr(kNoFallbackAttr);
          return;
        }
        if (!ConvertToFlexOp(op)) signalPassFailure();
      });
    }
    }  // namespace internal
    
    std::unique_ptr<OperationPass<func::FuncOp>> CreateFallbackToFlexOpsPass(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 12.2K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc

      // errors later on.  These ops were likely lifted out of the
      // tf_device.cluster in an earlier pass.
      module.walk([](Operation* op) {
        op->removeAttr(kXlaOutsideCompilationAttr);
        op->removeAttr(kXlaMapOutsideCompilationAttr);
      });
    
      if (failed(CheckPostconditions(module))) return signalPassFailure();
    }
    }  // namespace
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 68.3K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc

                                    PatternRewriter& rewriter) const override {
        const auto f_attr = mlir::dyn_cast<FlatSymbolRefAttr>(call_op.getFAttr());
        // removeAttr will return nullptr if no attribute was removed.
        if (!call_op->removeAttr(kQuantTraitAttrName) || !f_attr) {
          return failure();
        }
        if (!f_attr.getValue().starts_with(kCompositeFuncPrefix)) {
          return failure();
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 54.5K bytes
    - Viewed (0)
Back to top