Searched refs:SpaceToBatchOp (Results 1 – 2 of 2) sorted by relevance
224 class SpaceToBatchOp : public OpKernel { class226 explicit SpaceToBatchOp(OpKernelConstruction* context) : OpKernel(context) { in SpaceToBatchOp() function in tensorflow::SpaceToBatchOp268 SpaceToBatchOp<CPUDevice, T>);285 SpaceToBatchOp<GPUDevice, T>);
168 class SpaceToBatchOp : public XlaOpKernel { class170 explicit SpaceToBatchOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in SpaceToBatchOp() function in tensorflow::__anon9eb6bd6e0111::SpaceToBatchOp189 SpaceToBatchOp);