Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for TF_KernelBuilder_HostMemory (0.17 sec)

  1. tensorflow/c/kernels_test.cc

        TF_DeleteStatus(status);
      };
    
      TF_KernelBuilder* builder = TF_NewKernelBuilder(
          op_name, device_name, &MyCreateFunc, my_compute_func, &MyDeleteFunc);
      TF_KernelBuilder_HostMemory(builder, "input2");
      TF_KernelBuilder_HostMemory(builder, "output1");
      TF_Status* status = TF_NewStatus();
      TF_RegisterKernelBuilder(node_name, builder, status);
      EXPECT_EQ(TF_OK, TF_GetCode(status));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 50.4K bytes
    - Viewed (0)
Back to top