Searched refs:BatchToSpaceOp (Results 1 – 2 of 2) sorted by relevance
219 class BatchToSpaceOp : public OpKernel { class221 explicit BatchToSpaceOp(OpKernelConstruction* context) : OpKernel(context) { in BatchToSpaceOp() function in tensorflow::BatchToSpaceOp264 BatchToSpaceOp<CPUDevice, T>);281 BatchToSpaceOp<GPUDevice, T>);
166 class BatchToSpaceOp : public XlaOpKernel { class168 explicit BatchToSpaceOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in BatchToSpaceOp() function in tensorflow::__anon59c828ae0111::BatchToSpaceOp187 BatchToSpaceOp);