Searched refs:MklConvOp (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.cc | 417 class MklConvOp : public OpKernel { class
419 ~MklConvOp() {} in ~MklConvOp()
421 explicit MklConvOp(OpKernelConstruction* context) : OpKernel(context) { in MklConvOp() function in tensorflow::MklConvOp
1160 : public MklConvOp<Device, Tinput, Tfilter, Tbias, Toutput, Ttemp_output,
1164 : MklConvOp<Device, Tinput, Tfilter, Tbias, Toutput, Ttemp_output, in MklFusedConvOp()
1288 : public MklConvOp<Device, Tinput, Tfilter, Tbias, Toutput, Ttemp_output,
1293 : MklConvOp<Device, Tinput, Tfilter, Tbias, Toutput, Ttemp_output, in MklFusedDepthwiseConvOp()
1344 : public MklConvOp<Device, Tinput, qint8, Tbias, Toutput, Ttemp_output,
1360 : MklConvOp<Device, Tinput, qint8, Tbias, Toutput, Ttemp_output, int32, in MklQuantizedConv2DOp()
1377 MklConvOp<Device, Tinput, qint8, Tbias, Toutput, Ttemp_output, int32, in Compute()
[all …]
|