Searched refs:ConvBackpropFilterOp (Results 1 – 1 of 1) sorted by relevance
147 class ConvBackpropFilterOp : public XlaOpKernel { class149 explicit ConvBackpropFilterOp(OpKernelConstruction* ctx, int num_spatial_dims, in ConvBackpropFilterOp() function in tensorflow::__anon7c19f5bd0111::ConvBackpropFilterOp175 TF_DISALLOW_COPY_AND_ASSIGN(ConvBackpropFilterOp);178 class Conv2DBackpropFilterOp : public ConvBackpropFilterOp {181 : ConvBackpropFilterOp(ctx, /*num_spatial_dims=*/2, /*depthwise=*/false) { in Conv2DBackpropFilterOp()188 class Conv3DBackpropFilterOp : public ConvBackpropFilterOp {191 : ConvBackpropFilterOp(ctx, /*num_spatial_dims=*/3, /*depthwise=*/false) { in Conv3DBackpropFilterOp()198 class DepthwiseConv2DBackpropFilterOp : public ConvBackpropFilterOp {201 : ConvBackpropFilterOp(ctx, /*num_spatial_dims=*/2, /*depthwise=*/true) {} in DepthwiseConv2DBackpropFilterOp()