- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 73 for HLO (0.02 sec)
-
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-with-tf2xla-hlo-importer.mlir
} func.func @non_const_inputs(%arg0: tensor<2x2xf64>, %arg1: tensor<f64>, %arg2: tensor<2xi32>, %arg3: tensor<2xi32>, %arg4: tensor<2xi32>) -> tensor<6x5xf64> { // expected-remark@+1 {{compilation to HLO failed: INVALID_ARGUMENT: Input 2 to node `tf.XlaPad` with op XlaPad must be a compile-time constant.}}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 38.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.h
custom_legalization_passes); // Compiles a Graph from TF to HLO and adds the resulting HLO to the // XlaBuilder. This function adds HLO to a larger HLO computation, so // HLO-level inputs are supplied, and HLO-level outputs are produced. // xla_params is the HLO-level inputs and returns is the HLO-level outputs. // If unconditionally_use_output_shapes is true then the unregistered
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/lite/tf_tfl_translate_cl.cc
// NOLINTNEXTLINE opt<bool> import_hlo("import-hlo", llvm::cl::desc("Whether the input file is hlo file."), llvm::cl::init(false)); // NOLINTNEXTLINE opt<HloImportType> hlo_import_type( "hlo-import-type", llvm::cl::desc("The file type of the hlo."), llvm::cl::values(clEnumVal(proto, "Import hlo in proto binary format"),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 20:53:17 UTC 2024 - 7.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_tf_passes.td
// Declare passes used in xla_legalize_tf. include "mlir/Pass/PassBase.td" def LegalizeTF : Pass<"xla-legalize-tf", "ModuleOp"> { let summary = "Legalize from TF dialect's or HLO dialect's control flow."; let description = [{ Legalizes from TF dialect to HLO dialect. When allow_partial_conversion is false, emits an error if there is any operation that can't be legalized.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 21 17:44:14 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/passes.h
RewritePatternSet* patterns); /// Converts the provided Operation as well as all nested operations into HLO /// dialect using the conversion patterns registered by the HLO dialect. When /// allow_partial_conversion is false, emits an error if there is any operation /// that can't be legalized. /// When `tf2xla_fallback_device_type` is not `None`, also uses legalization
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_tfl_translate_cl.h
extern llvm::cl::opt<std::string> saved_model_exported_names; // Import HLO. enum HloImportType { proto, hlotxt, mlir_text }; extern llvm::cl::opt<bool> import_hlo; extern llvm::cl::opt<HloImportType> hlo_import_type; // enable_hlo_to_tf_conversion and disable_hlo_to_tfl_conversion are used to // control the HLO to TF and HLO to TFLite conversion while debugging an
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 20:53:17 UTC 2024 - 3.3K bytes - Viewed (0) -
tensorflow/compiler/jit/device_compiler_client.h
namespace tensorflow { template <typename ExecutableType, typename ClientType> class DeviceCompilerClient { public: DeviceCompilerClient() = default; virtual ~DeviceCompilerClient() = default; // Compiles `result` (HLO) to an `ExecutableType` using `ClientType` and // returns it. virtual StatusOr<std::unique_ptr<ExecutableType>> BuildExecutable( const XlaCompiler::Options& options,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/test_matchers_test.cc
} tensorflow::XlaCompilationResult CreateXlaComputationResult( const char* hlo_name) { auto result = tensorflow::XlaCompilationResult(); xla::HloModuleProto hlo; hlo.set_name(hlo_name); result.computation = std::make_shared<xla::XlaComputation>(hlo); return result; } TEST(TestUtil, ComputationContainsOk) { constexpr char arbitrary_hlo[] = "arbitrary_hlo"; auto result = CreateXlaComputationResult(arbitrary_hlo);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 20:29:34 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/python/jax_to_tfl_flatbuffer.cc
#include "tensorflow/compiler/mlir/lite/transforms/passes.h" #include "tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_config.h" #include "xla/service/hlo.pb.h" #include "xla/service/hlo_parser.h" #include "xla/translate/hlo_to_mhlo/hlo_to_mlir_hlo.h" #include "tensorflow/core/framework/graph.pb.h" #include "tensorflow/core/framework/graph_debug_info.pb.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 11 19:29:56 UTC 2024 - 8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf.cc
compilation_result.get()); if (combined_bridge_status.ok()) { VLOG(1) << "Successfully compiled MLIR computation to XLA HLO using " "Combined MLIR and XlaBuilder Bridge."; DumpHloCompilationResult("legalize_tf_combined_bridge.hlo", compilation_result.get()) .IgnoreError(); return *compilation_result; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 00:40:46 UTC 2024 - 6.8K bytes - Viewed (0)