Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for CompilableOp (0.22 sec)

  1. tensorflow/compiler/jit/compilability_check_util_test.cc

        ctx->SetOutput(0, ctx->Input(0));
      }
    };
    
    // Register the DummyCompilableOp kernel for CPU.
    REGISTER_OP("InputFloatOp").Output("o: float");
    REGISTER_OP("CompilableOp").Input("i: float").Output("o: float");
    REGISTER_XLA_OP(Name("CompilableOp").Device(DEVICE_CPU_XLA_JIT),
                    DummyCompilableOp);
    
    // Dummy op that is uncompilable in CPU.
    REGISTER_OP("MissingKernel").Input("i: float").Output("o: float");
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jun 10 12:32:39 UTC 2022
    - 22.3K bytes
    - Viewed (0)
Back to top