Searched refs:ThreadIDOp (Results 1 – 1 of 1) sorted by relevance
1383 class ThreadIDOp : public OpKernel { class1385 explicit ThreadIDOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in ThreadIDOp() function in tensorflow::__anon93d6499e0111::ThreadIDOp1395 REGISTER_KERNEL_BUILDER(Name("ThreadID").Device(DEVICE_CPU), ThreadIDOp);