Searched refs:ConvBackpropInputOp (Results 1 – 1 of 1) sorted by relevance
89 class ConvBackpropInputOp : public XlaOpKernel { class91 explicit ConvBackpropInputOp(OpKernelConstruction* ctx, int num_spatial_dims, in ConvBackpropInputOp() function in tensorflow::__anon7c19f5bd0111::ConvBackpropInputOp117 TF_DISALLOW_COPY_AND_ASSIGN(ConvBackpropInputOp);120 class Conv2DBackpropInputOp : public ConvBackpropInputOp {123 : ConvBackpropInputOp(ctx, /*num_spatial_dims=*/2, /*depthwise=*/false) {} in Conv2DBackpropInputOp()129 class Conv3DBackpropInputOp : public ConvBackpropInputOp {132 : ConvBackpropInputOp(ctx, /*num_spatial_dims=*/3, /*depthwise=*/false) {} in Conv3DBackpropInputOp()138 class DepthwiseConv2DBackpropInputOp : public ConvBackpropInputOp {141 : ConvBackpropInputOp(ctx, /*num_spatial_dims=*/2, /*depthwise=*/true) {} in DepthwiseConv2DBackpropInputOp()