Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for XlaOpKernel (0.25 sec)

  1. tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config.cc

    namespace mlir {
    namespace mhlo {
    
    namespace {
    
    // Returns ops that should use MLIR legalization.
    // All other ops not in this list should use XlaOpKernel.
    const llvm::DenseSet<mlir::TypeID>& MlirAlwaysOps() {
      // The static variable is a pointer in order to avoid destruction upon thread
      // termination.
      static const llvm::DenseSet<mlir::TypeID>* ops = new llvm::DenseSet<
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 04:08:35 UTC 2024
    - 21.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/compilability_check_util_test.cc

    constexpr char kUncompilableFunctionNodeTwoName[] = "n_d_uncompilable";
    
    // A dummy OpKernel for testing.
    class DummyCompilableOp : public XlaOpKernel {
     public:
      explicit DummyCompilableOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {}
      void Compile(XlaOpKernelContext* ctx) override {
        ctx->SetOutput(0, ctx->Input(0));
      }
    };
    
    // Register the DummyCompilableOp kernel for CPU.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jun 10 12:32:39 UTC 2022
    - 22.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/compilability_check_util.h

        bool allow_control_trigger = false;
    
        // Whether it is okay to "cluster" Assert and CheckNumerics by simply
        // removing them (they're not removed during clustering, but their
        // XlaOpKernel is a no-op kernel).  We avoid auto-clustering these ops so
        // that the user is not surprised when XLA is implicitly enabled. If the
        // user explicitly specifies to use XLA, it is fine to resort to a dummy
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 14.9K bytes
    - Viewed (0)
Back to top