- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for LegalizeMlirToHlo (0.34 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/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/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)