Searched refs:SpaceToBatchNDOp (Results 1 – 7 of 7) sorted by relevance
208 class SpaceToBatchNDOp : public OpKernel { class210 explicit SpaceToBatchNDOp(OpKernelConstruction* context) in SpaceToBatchNDOp() function in tensorflow::SpaceToBatchNDOp263 SpaceToBatchNDOp<CPUDevice, T>); \280 SpaceToBatchNDOp<GPUDevice, T>); \
148 class SpaceToBatchNDOp : public XlaOpKernel { class150 explicit SpaceToBatchNDOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in SpaceToBatchNDOp() function in tensorflow::__anon84839c050111::SpaceToBatchNDOp166 SpaceToBatchNDOp);
161 if (!llvm::isa<TF::SpaceToBatchNDOp>(prev_op)) return failure(); in matchAndRewrite()163 TF::SpaceToBatchNDOp stb_op = llvm::cast<TF::SpaceToBatchNDOp>(prev_op); in matchAndRewrite()
231 TypeID::get<TF::SpaceToBatchNDOp>(), in IsOpAllowedTf2XlaFallback()
803 : RewritePattern(SpaceToBatchNDOp::getOperationName(), in LowerSpaceToBatchNDOp()821 auto op = cast<SpaceToBatchNDOp>(src_op); in matchAndRewrite()
1196 static LogicalResult Verify(SpaceToBatchNDOp op) { in Verify()1298 LogicalResult SpaceToBatchNDOp::inferReturnTypes( in inferReturnTypes()
1817 auto tf_s2b_op = cast<TF::SpaceToBatchNDOp>(op); in matchAndRewrite()