- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for lower_to_xla_hlo (0.3 sec)
-
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.h
llvm::MutableArrayRef<std::unique_ptr<mlir::Pass>> custom_legalization_passes, llvm::StringRef module_name = llvm::StringRef(), bool lower_to_xla_hlo = true); // Runs MLIR Bridge on a serialized MLIR module. // // If lower_to_xla_hlo is true then compiles down into XLA HLO, generates all // accompanying metadata and stores them in CompilationResult.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 10.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
void AddLegalizationPasses(mlir::OpPassManager& pm, bool legalize_chlo, llvm::StringRef device_type, bool enable_op_fallback, bool lower_to_xla_hlo) { if (lower_to_xla_hlo) { // Lower TF quant ops and types to MHLO int. mlir::quant::stablehlo::AddQuantizationLoweringPasses(pm); pm.addPass(mlir::mhlo::createLegalizeTFPass( legalize_chlo,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_mlir.cc
// enable logging. constexpr char kBridgeComponent[] = "TFXLABridge"; using tpu::MlirToHloArgs; using tpu::ShardingAndIndex; absl::StatusOr<std::string> CompileFromMlirToXlaHlo( bool lower_to_xla_hlo, const MlirToHloArgs& computation, const tpu::TPUCompileMetadataProto& metadata, llvm::StringRef device_type, const XlaShapeLayoutHelpers::ShapeDeterminationFns& shape_determination_fns,
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/tf2xla/api/v1/compile_mlir_util_test.cc
/*enable_op_fallback=*/false, /*custom_legalization_passes*/ {}, /*lower_to_xla_hlo=*/false); std::string pass_description; llvm::raw_string_ostream raw_stream(pass_description); pass_manager.printAsTextualPipeline(raw_stream);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 25 19:54:38 UTC 2024 - 9.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_mlir.h
// then those passes include all the Legalization to XLA HLO which is returned // in the compilation_result. absl::StatusOr<std::string> CompileFromMlirToXlaHlo( bool lower_to_xla_hlo, const tpu::MlirToHloArgs& computation, const tpu::TPUCompileMetadataProto& metadata, llvm::StringRef device_type, const XlaShapeLayoutHelpers::ShapeDeterminationFns& shape_determination_fns,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 20:29:34 UTC 2024 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_to_hlo.cc
"Combined MLIR Tf2Xla Bridge."; absl::StatusOr<std::string> mlir_compilation = internal::CompileFromMlirToXlaHlo( /*lower_to_xla_hlo=*/false, computation, metadata, device_type, shape_determination_fns, use_tuple_args, compilation_result, custom_legalization_passes, arg_shapes, arg_core_mapping, per_core_arg_shapes);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 20:29:34 UTC 2024 - 3.7K bytes - Viewed (0)