- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for LegalizeMlirToHlo (0.25 sec)
-
tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf.cc
return; } switch (computation.index()) { case 0: VLOG(2) << "LegalizeMlirToHlo with MLIR computation input: " << std::get<0>(computation).mlir_module; break; case 1: { auto input = std::get<1>(computation); Graph g(input.flib_def); VLOG(2) << "LegalizeMlirToHlo with FLIB computation input: " << DumpGraphToFile(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 00:40:46 UTC 2024 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.h
// result shapes. // custom_legalization_passes: passes to run before the default TF legalization // passes for backend-specific ops. ABSL_DEPRECATED("Use v2/legalize_tf.h::LegalizeMlirToHlo instead.") Status ConvertMLIRToXlaComputation( mlir::ModuleOp module_op, llvm::StringRef device_type, xla::XlaComputation* xla_computation, bool use_tuple_args, bool enable_op_fallback, bool return_tuple,
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/v2/legalize_tf.h
// arg_core_mapping - Which args go on which cores. // per_core_arg_shapes - For each core, the shapes for each argument. // client - The Xla Compilation client. absl::StatusOr<tensorflow::XlaCompilationResult> LegalizeMlirToHlo( const std::variant<tpu::MlirToHloArgs, tpu::FunctionToHloArgs>& computation, const tpu::TPUCompileMetadataProto& metadata, bool use_tuple_args, llvm::StringRef device_type,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 07:32:57 UTC 2024 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf_test.cc
std::vector<ShardingAndIndex> arg_core_mapping; std::vector<std::vector<xla::Shape>> per_core_arg_shapes; std::vector<std::unique_ptr<mlir::Pass>> custom_legalization_passes; return LegalizeMlirToHlo(mlir_to_hlo_args, metadata_proto, use_tuple_args, /*device_type=*/"XLA_TPU_JIT", custom_legalization_passes,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 23:59:33 UTC 2024 - 16.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/legalize_tf_quant_test.cc
std::vector<std::vector<xla::Shape>> per_core_arg_shapes; std::vector<std::unique_ptr<mlir::Pass>> custom_legalization_passes; TF_EXPECT_OK(tensorflow::tf2xla::v2::LegalizeMlirToHlo( mlir_to_hlo_args, metadata_proto, use_tuple_args, /*device_type=*/"XLA_TPU_JIT", custom_legalization_passes, /*shape_determination_fns=*/{}, arg_shapes,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 18:43:55 UTC 2024 - 7.2K bytes - Viewed (0)