- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for CompilableOp (0.09 sec)
-
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)