- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 104 for HLO (0.02 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/utils.cc
OpBuilder* builder) { return builder->create<ConstantOp>(loc, hlo::getScalarOfType(ty, raw_value)); } ConstantOp GetScalarNegZeroOfType(Type ty, Location loc, OpBuilder* builder) { return builder->create<ConstantOp>(loc, hlo::getScalarNegZeroOfType(ty)); } DenseIntElementsAttr GetI64ElementsAttr(ArrayAttr attr) { RankedTensorType ty =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/compile_mlir_util/constant-folding-hook.mlir
// RUN: tf-mlir-translate -mlir-tf-to-hlo-text %s -tf-input-shapes=: -tf-xla-emit-use-tuple-args -tf-xla-emit-return-tuple | FileCheck %s // RUN: tf-mlir-translate -mlir-tf-to-hlo-text %s -tf-input-shapes=: | FileCheck -check-prefix=NO_TUPLES %s // RUN: tf-mlir-translate -mlir-tf-to-hlo-text-via-builder %s -tf-input-shapes=: | FileCheck -check-prefix=NO_TUPLES %s module attributes {tf.versions = {producer = 179 : i32}} { func.func @main() -> (tensor<0xi32>, tensor<0xi32>) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 23 18:56:13 UTC 2022 - 1.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compilation_cache.proto
XlaSerializedCacheKey key = 1; // The computation (HLO) that compilation was done for. It is correlated to // the input TF graph so we can use it to fingerprint the compiled binary. We // serialize this rather than the input graphdef because it provides a // stronger guarantee over what bindings are needed between the HLO and // calling TF graph. xla.HloModuleProto hlo_module = 2;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 1.6K 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/jit/tests/device_compiler_serialize_test.cc
// Run again but these should all fail, because the persistent cache entries' // HLO modules have been altered. for (int b = 1; b < 4; ++b) { auto status = ExecuteWithBatch(graph, b); EXPECT_FALSE(status.ok()); EXPECT_TRUE( absl::StrContains(status.message(), "Serialized HLO does not match.")); } } } // namespace } // namespace tensorflow
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 15 15:13:38 UTC 2023 - 2.7K 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) -
tensorflow/compiler/mlir/tensorflow/utils/tf_xla_mlir_translate.cc
static mlir::TranslateFromMLIRRegistration MlirTfToHloTextTranslate( "mlir-tf-to-hlo-text", "mlir-tf-to-hlo-text", tensorflow::MlirTfToHloTextTranslateFunction, tensorflow::RegisterMlirInputDialects); static mlir::TranslateFromMLIRRegistration MlirTfToHloTextViaBuilderTranslate( "mlir-tf-to-hlo-text-via-builder", "mlir-tf-to-hlo-text-via-builder", tensorflow::MlirTfToHloTextViaBuilderTranslateFunction,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf.h
#include "tsl/platform/statusor.h" namespace tensorflow { namespace tf2xla { namespace v2 { // Legalizes the given mlir::Module into XLA HLO. If successful, returns the // compiled XLA HLO. V1 of the tf2xla uses MLIR whereas V0 does not use MLIR. // // Inputs: // computation - The MLIR module op. It currently takes in
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/tensorflow/tests/compile_mlir_util/argument-sharding-invalid.mlir
// RUN: not tf-mlir-translate -mlir-tf-to-hlo-text %s -tf-input-shapes=128,10 -tf-xla-emit-use-tuple-args -tf-xla-emit-return-tuple 2>&1 | FileCheck %s module attributes {tf.versions = {producer = 179 : i32}} { func.func @main(%arg0: tensor<128x8xf32> {mhlo.sharding = "bad_sharding"}) { func.return } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 364 bytes - Viewed (0)