Searched refs:DepthwiseConv2dNativeOp (Results 1 – 6 of 6) sorted by relevance
269 class DepthwiseConv2dNativeOp : public BinaryOp<T> { class271 explicit DepthwiseConv2dNativeOp(OpKernelConstruction* context) in DepthwiseConv2dNativeOp() function in tensorflow::DepthwiseConv2dNativeOp489 TF_DISALLOW_COPY_AND_ASSIGN(DepthwiseConv2dNativeOp);495 DepthwiseConv2dNativeOp<CPUDevice, T>)508 DepthwiseConv2dNativeOp<GPUDevice, T>)517 : public DepthwiseConv2dNativeOp<GPUDevice, T> {520 : DepthwiseConv2dNativeOp<GPUDevice, T>(context) { in DepthwiseConv2dGroupedConvOp()
33 ConvertTFDilatedConvOp<TF::DepthwiseConv2dNativeOp>>( in runOnFunction()
439 TF::DepthwiseConv2dNativeOp> {442 TF::DepthwiseConv2dNativeOp>;1465 ConvertTFDilatedConvOp<TF::DepthwiseConv2dNativeOp>>(ctx); in runOnFunction()
178 rewriter.replaceOpWithNewOp<DepthwiseConv2dNativeOp>( in CreateConvOp()
846 auto tf_dwconv2d_op = cast<TF::DepthwiseConv2dNativeOp>(op); in matchAndRewrite()
1288 ConvertConvOp<TF::DepthwiseConv2dNativeOp, /*num_spatial_dims=*/2,