Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for LegalizeMlirToHlo (0.27 sec)

  1. 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)
  2. 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)
Back to top