Home
last modified time | relevance | path

Searched refs:DepthwiseConv2dNativeBackpropInputOp (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Ddepthwise_conv_grad_op.cc554 class DepthwiseConv2dNativeBackpropInputOp : public OpKernel { class
556 explicit DepthwiseConv2dNativeBackpropInputOp(OpKernelConstruction* context) in DepthwiseConv2dNativeBackpropInputOp() function in tensorflow::DepthwiseConv2dNativeBackpropInputOp
708 TF_DISALLOW_COPY_AND_ASSIGN(DepthwiseConv2dNativeBackpropInputOp);
715 DepthwiseConv2dNativeBackpropInputOp<CPUDevice, T>);
732 DepthwiseConv2dNativeBackpropInputOp<GPUDevice, T>)
742 : public DepthwiseConv2dNativeBackpropInputOp<GPUDevice, T> {
745 : DepthwiseConv2dNativeBackpropInputOp<GPUDevice, T>(context) { in DepthwiseConv2dGroupedConvBackpropInputOp()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf_with_tf2xla.cc130 TypeID::get<TF::DepthwiseConv2dNativeBackpropInputOp>(), in IsOpAllowedTf2XlaFallback()