Searched refs:SpaceToBatchOp (Results 1 – 3 of 3) 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::__anon84839c050111::SpaceToBatchOp189 SpaceToBatchOp);
232 TypeID::get<TF::SpaceToBatchOp>(), in IsOpAllowedTf2XlaFallback()