- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for OpKernel (0.16 sec)
-
tensorflow/c/eager/c_api_test.cc
.Doc(R"doc(Test communication op throwing Unavailable error.)doc"); // Kernel that throws an Unavailable error. class TestUnavailableErrorOp : public tensorflow::OpKernel { public: explicit TestUnavailableErrorOp(tensorflow::OpKernelConstruction* ctx) : tensorflow::OpKernel(ctx) {} void Compute(tensorflow::OpKernelContext* ctx) override { ctx->SetStatus(tensorflow::errors::Unavailable("Test error.")); } };
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 20:50:20 GMT 2023 - 94.6K bytes - Viewed (1) -
tensorflow/c/c_api_test.cc
TF_DeleteBuffer(api_def_buf); TF_DeleteApiDefMap(api_def_map); TF_DeleteBuffer(op_list_buf); } class DummyKernel : public tensorflow::OpKernel { public: explicit DummyKernel(tensorflow::OpKernelConstruction* context) : OpKernel(context) {} void Compute(tensorflow::OpKernelContext* context) override {} }; // Test we can query kernels REGISTER_OP("TestOpWithSingleKernel")
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 96.9K bytes - Viewed (3)