Searched refs:BatchKernel (Results 1 – 1 of 1) sorted by relevance
780 class BatchKernel : public AsyncOpKernel { class782 explicit BatchKernel(OpKernelConstruction* c) : AsyncOpKernel(c) { in BatchKernel() function in tensorflow::BatchKernel856 REGISTER_KERNEL_BUILDER(Name("Batch").Device(DEVICE_CPU), BatchKernel);