- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 18 for xla_tpu_jit (0.16 sec)
-
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/api/v1/compile_mlir_util.h
// . Legalizes the operations to XLA HLO operations. // . Canonicalizes the XLA HLO operations. // // device_type: XLA JIT device to use for compilation such as "XLA_CPU_JIT", // "XLA_GPU_JIT" or "XLA_TPU_JIT". // use_tuple_args: when this is true, always create a tuple argument for the // entry computation. // enable_op_fallback: when this is true, prefer tf2xla fallback kernels over // MLIR
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/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/jit/BUILD
], deps = [ ":xla_cpu_device", ":xla_cpu_jit", "//tensorflow/compiler/plugin", ] + if_cuda_or_rocm([ ":xla_gpu_device", ":xla_gpu_jit", ]) + if_with_tpu_support([ ":xla_tpu_device", ":xla_tpu_jit", ]), alwayslink = 1, ) cc_library( name = "xla_cpu_jit", visibility = ["//visibility:public"], deps = [
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 61.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf_test.cc
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, /*shape_determination_fns=*/{}, arg_shapes, &arg_core_mapping, &per_core_arg_shapes, client); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 23:59:33 UTC 2024 - 16.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter_test.cc
explicit Tf2XlaRewriterTestPeer(mlir::Operation* op) : op_builder_(op), empty_rewriter_(op_builder_), tf2xla_rewriter_(op, empty_rewriter_, /*device_type=*/"XLA_CPU_JIT") {} absl::StatusOr<TupleOp> ImportXlaComputationIntoModule( XlaComputation& computation) { return tf2xla_rewriter_.ImportXlaComputation(computation); } private: OpBuilder op_builder_;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 11.7K bytes - Viewed (0) -
tensorflow/compiler/jit/pjrt_base_device.h
std::string device_name; // The index of the device. int device_ordinal = -1; // The name of the compilation device, also referred to as jit_device_type. // (e.g., "XLA_CPU_JIT"); std::string compilation_device_name; // A vector of ShapeDeterminationFn (i.e., a bundle of LayoutSelectionFn, // ShapeRepresentationFn). Each bundle describes how the on-host shapes of
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 4K bytes - Viewed (0)