- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 293 for runOnOperation (0.17 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/verify_clustering_pass.cc
using mlir::Operation; using mlir::WalkResult; class VerifyClusteringPass : public impl::VerifyClusteringPassBase<VerifyClusteringPass> { public: void runOnOperation() override; }; void VerifyClusteringPass::runOnOperation() { Operation* func_op = getOperation(); auto walk_result = func_op->walk([&](Operation* op) { if (!tensorflow::tf2xla::internal::IsInBridgeAcceptableDialects(op)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 08 20:33:37 UTC 2023 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/rewrite_quantized_io.cc
StringRef getDescription() const final { return "Replaces operands and results that has quantized type with their " "storage types."; } void runOnOperation() override; }; void RewriteQuantizedIOPass::runOnOperation() { ModuleOp module = getOperation(); OpBuilder builder(module); module.walk([&](func::FuncOp func) { Block& block = func.front();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 01:19:25 UTC 2023 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_lowering_pass.cc
public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(CompositeLoweringPass); void runOnOperation() override; }; #include "tensorflow/compiler/mlir/lite/stablehlo/transforms/generated_composite_lowering.inc" void CompositeLoweringPass::runOnOperation() { RewritePatternSet patterns(&getContext()); populateWithGenerated(patterns); ConversionTarget target(getContext());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:16:05 UTC 2024 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_nnapi.cc
} void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect>(); } void runOnOperation() override; }; void DeviceTransformNNAPIPass::runOnOperation() { auto func = getOperation(); auto* ctx = &getContext(); NNAPIHardware nnapi_hardware; RewritePatternSet patterns = nnapi_hardware.GetTransformations(ctx);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 07 18:43:51 UTC 2022 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/input_lowering_metrics_pass.cc
constexpr char kDynamicFunctionName[] = "kDynamicFunction"; class InputMetricsLoweringPass : public impl::InputLoweringMetricsPassBase<InputMetricsLoweringPass> { public: void runOnOperation() override; }; void InputMetricsLoweringPass::runOnOperation() { bool has_dynamic_op = false; Operation* func_op = getOperation(); func_op->walk([&](Operation* op) { auto abstractOp = op->getRegisteredInfo();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 08:55:35 UTC 2024 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/infeed_ops_xla_adjust_layout.cc
#include "tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_tf_passes.h.inc" class InfeedsOpsXlaAdjustLayout : public impl::InfeedOpsXlaAdjustLayoutBase<InfeedsOpsXlaAdjustLayout> { public: void runOnOperation() override; private: static void runOnInfeedOp(::mlir::mhlo::InfeedOp op) { OpBuilder builder(op.getContext()); SmallVector<Type> result_types(op.getResultTypes().begin(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_launch_func_to_tf_call.cc
struct ConvertLaunchFuncToTFCallPass : public impl::ConvertLaunchFuncToTFCallPassBase< ConvertLaunchFuncToTFCallPass> { void runOnOperation() override; }; void ConvertLaunchFuncToTFCallPass::runOnOperation() { ModuleOp module = getOperation(); module.walk([&](tf_device::LaunchFuncOp launch) { OpBuilder builder(launch); auto call_op = builder.create<TF::PartitionedCallOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 31 21:08:09 UTC 2023 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/freeze_saved_model_assets.cc
FreezeAssetsPass(const FreezeAssetsPass& pass) {} explicit FreezeAssetsPass(std::string saved_model_dir) { this->saved_model_dir = saved_model_dir; } void runOnOperation() override; private: std::string saved_model_dir; }; void FreezeAssetsPass::runOnOperation() { auto module = getOperation(); if (!tf_saved_model::HasTfSavedModelSemantics(module)) { return; } SymbolTable symbol_table(module);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jan 30 01:12:09 UTC 2023 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantize_composite_functions.cc
const bool enable_per_channel_quantized_weight) { enable_per_channel_quantized_weight_ = enable_per_channel_quantized_weight; } private: void runOnOperation() override; }; void QuantizeCompositeFunctionsPass::runOnOperation() { MLIRContext& ctx = getContext(); PassManager pm(&ctx); // Intermediate output from QuantizePass will have quantized ops
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 02:59:01 UTC 2024 - 4.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_partitioned_op_conversion.cc
#include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc" struct TPUPartitionedOpConversionPass : public impl::TPUPartitionedOpConversionPassBase< TPUPartitionedOpConversionPass> { void runOnOperation() override; }; template <typename T> LogicalResult ReplacePartitionedOp(IntegerAttr num_cores_per_replica, T op) { constexpr bool is_input = std::is_same_v<std::decay_t<T>, TF::TPUPartitionedInputOp>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.1K bytes - Viewed (0)