Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for XLA_CPU_JIT (0.67 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
Back to top