- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 14 for xla_tpu_jit (0.17 sec)
-
tensorflow/compiler/mlir/tf2xla/api/v2/device_type.proto
syntax = "proto2"; package tensorflow.tf2xla.v2; // The requested device type to compile for. enum DeviceType { DEVICE_TYPE_UNSPECIFIED = 0; XLA_TPU_JIT = 1; XLA_CPU_JIT = 2; XLA_GPU_JIT = 3;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Sep 15 15:50:12 UTC 2023 - 204 bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util_test.cc
failed_legalization, arg_shapes, /*device_type=*/"XLA_TPU_JIT", /*use_tuple_args=*/true, /*enable_op_fallback=*/false, /*shape_determination_fns=*/{}, &compilation_result); EXPECT_FALSE(status.ok()); EXPECT_EQ(count.Delta("tf.DoesntExist", "Unknown"), 1); } TEST(CompileMlirUtil, CreatesPipeline) { OpPassManager pass_manager; llvm::StringRef device_type = "XLA_CPU_JIT";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 25 19:54:38 UTC 2024 - 9.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_mlir_test.cc
auto compilation_result = std::make_unique<XlaCompilationResult>(); return CompileFromMlirToXlaHlo( compile_to_xla_hlo, mlir_to_hlo_args, metadata_proto, /*device_type=*/"XLA_TPU_JIT", /*shape_determination_fns=*/{}, use_tuple_args, compilation_result.get(), custom_legalization_passes, arg_shapes, &arg_core_mapping, &per_core_arg_shapes); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 20:29:34 UTC 2024 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/mlir_pass_instrumentation_test.cc
auto compilation_result = tensorflow::XlaCompilationResult(); TF_EXPECT_OK(tensorflow::CompileSerializedMlirToXlaHlo( legalization, arg_shapes, /*device_type=*/"XLA_TPU_JIT", /*use_tuple_args=*/true, /*enable_op_fallback=*/false, /*shape_determination_fns=*/{}, &compilation_result) .status());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 19 22:54:26 UTC 2023 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_to_hlo_test.cc
auto compilation_result = std::make_unique<XlaCompilationResult>(); return LegalizeTfToHlo(mlir_to_hlo_args, metadata_proto, use_tuple_args, /*device_type=*/"XLA_TPU_JIT", /*shape_determination_fns=*/{}, arg_shapes, &arg_core_mapping, &per_core_arg_shapes, custom_legalization_passes, client,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 20:29:34 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/lower_cluster_to_runtime_ops_test.cc
.ok()); EXPECT_EQ( compilation_status.Delta(mlir::TF::kMlirPh1BridgeCounterReplicated, mlir::TF::kMlirPh1BridgeCounterV2, "XLA_TPU_JIT", "fallback_disabled", "failure"), 1); } TEST_F(LowerClusterToRuntimeOpsTest, DumpsPipelinePasses) { std::vector<std::string> files;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:44:37 UTC 2024 - 6.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/legalize_tf_quant_test.cc
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, &arg_core_mapping, &per_core_arg_shapes, client) .status()); } };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 18:43:55 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/tfxla_device_specific_transformations_gpu.mlir
// RUN: tf-opt "--tfxla-device-specific-transforms=device-type=XLA_GPU_JIT" -verify-diagnostics -split-input-file %s | FileCheck -dump-input=fail %s module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 1399 : i32}} { // CHECK-LABEL: stateless_op func.func @stateless_op() -> tensor<i32> { // CHECK: %cst = "tf.Const"() <{value = dense<1> : tensor<i32>}> : () -> tensor<i32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 515 bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/tfxla_device_specific_transformations_cpu.mlir
// RUN: tf-opt "--tfxla-device-specific-transforms=device-type=XLA_CPU_JIT" -verify-diagnostics -split-input-file %s | FileCheck -dump-input=fail %s module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 1399 : i32}} { // CHECK-LABEL: stateless_op func.func @stateless_op() -> tensor<i32> { // CHECK: %cst = "tf.Const"() <{value = dense<1> : tensor<i32>}> : () -> tensor<i32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 515 bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/mlprogram.cc
pm.addPass(mlir::createCanonicalizerPass()); pm.addPass(mlir::createSymbolDCEPass()); pm.addPass(mlir::TF::CreateTFShapeInferencePass()); llvm::StringRef tf2xla_fallback_device_type = "XLA_CPU_JIT"; pm.addPass(mlir::mhlo::createLegalizeTFPass( /*legalize_chlo=*/true, tf2xla_fallback_device_type, /*prefer_tf2xla=*/false)); pm.addPass(mlir::TF::CreateStripTfAttributesPass());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 09 22:39:15 UTC 2024 - 3.3K bytes - Viewed (0)