Searched refs:UnbatchKernel (Results 1 – 1 of 1) sorted by relevance
1052 class UnbatchKernel : public AsyncOpKernel { class1054 explicit UnbatchKernel(OpKernelConstruction* c) : AsyncOpKernel(c) { in UnbatchKernel() function in tensorflow::UnbatchKernel1087 REGISTER_KERNEL_BUILDER(Name("Unbatch").Device(DEVICE_CPU), UnbatchKernel);