Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_ops.cc67 class Conv2DOp : public ConvOp { class
69 explicit Conv2DOp(OpKernelConstruction* ctx) in Conv2DOp() function in tensorflow::__anon1313cb460111::Conv2DOp
72 REGISTER_XLA_OP(Name("Conv2D"), Conv2DOp);
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops.cc442 class Conv2DOp : public BinaryOp<T> { class
444 explicit Conv2DOp(OpKernelConstruction* context) : BinaryOp<T>(context) { in Conv2DOp() function in tensorflow::Conv2DOp
530 TF_DISALLOW_COPY_AND_ASSIGN(Conv2DOp);
536 Conv2DOp<CPUDevice, T>);
961 Conv2DOp<GPUDevice, Eigen::half>);
964 Conv2DOp<GPUDevice, float>);
967 Conv2DOp<GPUDevice, double>);