- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for XLA_CPU_JIT (0.67 sec)
-
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util_test.cc
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"; CreateConvertMlirToXlaHloPipeline(pass_manager, device_type, /*enable_op_fallback=*/false, /*custom_legalization_passes*/ {});
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/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) -
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) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/tf_stablehlo_pass.cc
mhlo::PopulateLegalizeTfPatterns(context, &patterns); TF::PopulateTFLoweringBeforeHLOPatterns(context, &patterns); mhlo::Tf2XlaTypeConverter converter; mhlo::PopulateLegalizeTfWithTf2XlaPatterns( "XLA_CPU_JIT", patterns, context, converter, /*prefer_tf2xla=*/false); stablehlo::StablehloToHloTypeConverter hlo_converter; chlo::populateChloToHloPatterns(context, &hlo_converter, &patterns);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_registration.cc
static constexpr char kMlirToGraphCompilationCheckName[] = "mlir-to-graph-compilation-check"; // Use CPU arbitrarily in order to check that a graph compiles at all static constexpr char kArbitraryDeviceName[] = "XLA_CPU_JIT"; namespace { inline absl::string_view StringRefToView(llvm::StringRef ref) { return {ref.data(), ref.size()}; } } // namespace static OwningOpRef<mlir::ModuleOp> GraphdefToMlirTranslateFunction(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 7.8K bytes - Viewed (0)