Searched refs:Conv2DBackpropInputOp (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_ops.cc | 132 class Conv2DBackpropInputOp : public ConvBackpropInputOp { class
134 explicit Conv2DBackpropInputOp(OpKernelConstruction* ctx) in Conv2DBackpropInputOp() function in tensorflow::__anon96cce0f30111::Conv2DBackpropInputOp
140 Conv2DBackpropInputOp);
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/tensorflow/ |
| D | fallback_to_flex_ops.cc | 88 TF::Conv2DBackpropInputOp::getOperationName().str(), in QuantizableOpsInLegacyMode()
121 TF::Conv2DBackpropInputOp::getOperationName().str(), in QuantizableOpsInDefaultMode()
142 TF::Conv2DBackpropInputOp, in IsFusibleWithBiasOp()
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | conv_spmd_expander.cc | 384 if (llvm::isa<mlir::TF::Conv2DBackpropInputOp>(op)) in ExpandOp()
386 *output_layout, llvm::cast<mlir::TF::Conv2DBackpropInputOp>(op)); in ExpandOp()
439 } else if (llvm::isa<mlir::TF::Conv2DBackpropInputOp, in ComputeLayoutForward()
448 if (llvm::isa<mlir::TF::Conv2DBackpropInputOp, in ComputeLayoutForward()
498 } else if (llvm::isa<mlir::TF::Conv2DBackpropInputOp, in ComputeLayoutBackward()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_input_ops.cc | 489 Conv2DBackpropInputOp<GPUDevice, double>);
494 Conv2DBackpropInputOp<GPUDevice, float>);
499 Conv2DBackpropInputOp<GPUDevice, Eigen::half>);
504 Conv2DBackpropInputOp<GPUDevice, int32>);
|
| D | conv_grad_input_ops.h | 354 class Conv2DBackpropInputOp : public OpKernel {
356 explicit Conv2DBackpropInputOp(OpKernelConstruction* context)
484 TF_DISALLOW_COPY_AND_ASSIGN(Conv2DBackpropInputOp);
815 Conv2DBackpropInputOp<CPUDevice, T>);
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 1924 LogicalResult Conv2DBackpropInputOp::verify() { in verify()
1925 Conv2DBackpropInputOp op = *this; in verify()
1949 LogicalResult Conv2DBackpropInputOp::UpdateDataFormat(StringRef data_format) { in UpdateDataFormat()
1974 StringRef Conv2DBackpropInputOp::GetOptimalLayout( in GetOptimalLayout()
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | spmd_expanders.cc | 217 REGISTER_SPMD(Conv2DBackpropInput, TF::Conv2DBackpropInputOp, ConvSPMDExpander);
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_with_tf2xla.cc | 309 TypeID::get<TF::Conv2DBackpropInputOp>(), in IsOpAllowedTf2XlaPreferred()
|
| D | legalize_tf.cc | 5079 ConvertConvBackpropInputOp<TF::Conv2DBackpropInputOp,
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 931 auto tf_conv_op = cast<TF::Conv2DBackpropInputOp>(op); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 780 rewriter.replaceOpWithNewOp<Conv2DBackpropInputOp>( in matchAndRewrite()
|