- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for DummyCompilableOp (0.12 sec)
-
tensorflow/compiler/jit/compilability_check_util_test.cc
// 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. REGISTER_OP("InputFloatOp").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)