- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for skip_resize (0.34 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/tf_stablehlo_pass.cc
bool skip_quantization_ops, bool skip_resize, bool skip_partitioned_calls) { TFToStablehloOptions options; options.skip_quantization_ops = skip_quantization_ops; options.skip_resize = skip_resize; options.skip_partitioned_calls = skip_partitioned_calls; PopulateLegalizeTFToStablehloPipeline(pm, options); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/transforms.h
// Adds all the necessary passes to lower a TF module to StableHLO. // `skip_resize` enables or disables skipping conversion of tf.ResizeBilinear // and tf.ResizeNearestNeighbor ops. // `smuggle_disallowed_ops` enables or disables converting disallowed ops // like tf.ResizeBilinear or tf.ResizeNearestNeighbor to mhlo.custom_call ops. void AddTFToStablehloPasses(OpPassManager& pm, bool skip_resize, bool smuggle_disallowed_ops);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 01:08:27 UTC 2024 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/transforms.cc
#include "tensorflow/compiler/mlir/tf2xla/transforms/passes.h" #include "xla/mlir_hlo/mhlo/transforms/passes.h" namespace mlir { namespace odml { void AddTFToStablehloPasses(OpPassManager& pm, bool skip_resize, bool smuggle_disallowed_ops) { pm.addPass(CreateRenameEntrypointToMainPass()); // if the input is a call_xla_module, then unwrap the content
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 04:34:23 UTC 2024 - 5.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/tf_stablehlo_pass.h
// Adds passes which transform TF Ops to StableHLO Ops. void AddLegalizeTFToStablehloPasses(OpPassManager& pm, bool skip_quantization_ops, bool skip_resize, bool skip_partitioned_calls); } // namespace odml } // namespace mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 08 20:05:12 UTC 2024 - 1.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/odml_to_stablehlo.cc
opt<bool> skip_checks("skip-checks", llvm::cl::desc("Skip checking for disallowed ops."), llvm::cl::Optional, llvm::cl::init(false)); // NOLINTNEXTLINE opt<bool> skip_resize( "skip-resize", llvm::cl::desc( "Skip converting tf.ResizeBilinear and tf.ResizeNearestNeighbor ops."), llvm::cl::Optional, llvm::cl::init(true)); // NOLINTNEXTLINE
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:16:49 UTC 2024 - 14.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/stablehlo/quantization.cc
pm.addPass(mlir::TF::CreateTFShapeInferencePass()); mlir::odml::AddLegalizeTFToStablehloPasses(pm, /*skip_quantization_ops=*/true, /*skip_resize=*/false, /*skip_partitioned_calls=*/false); pm.addNestedPass<mlir::func::FuncOp>( mlir::quant::stablehlo::createRemoveShardingCustomCallPass());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 10:49:12 UTC 2024 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/quantize_preprocess.cc
// TF -> StableHLO legalization. // Skip StatefulPartitionedCall to preserve aliased functions. mlir::odml::AddLegalizeTFToStablehloPasses(pm, /*skip_quantization_ops=*/true, /*skip_resize=*/false, /*skip_partitioned_calls=*/true); // StableHLO -> MHLO legalization for MHLO optimization. pm.addPass(mlir::mhlo::createStablehloLegalizeToHloPass());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 12:49:45 UTC 2024 - 9.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc
} if (failed(pass_manager.run(module))) { return status_handler.ConsumeStatus(); } } pass_manager.clear(); mlir::odml::AddTFToStablehloPasses(pass_manager, /*skip_resize=*/true, /*smuggle_disallowed_ops=*/true); // Print out a detailed report of non-converted stats. pass_manager.addPass(mlir::odml::createPrintOpStatsPass(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 23.8K bytes - Viewed (0)