Searched refs:MklConvFwdPrimitive (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.cc | 98 class MklConvFwdPrimitive : public MklPrimitive { class
100 explicit MklConvFwdPrimitive(const MklConvFwdParams& convFwdDims) in MklConvFwdPrimitive() function in tensorflow::MklConvFwdPrimitive
107 ~MklConvFwdPrimitive() {} in ~MklConvFwdPrimitive()
323 static MklConvFwdPrimitive<Tinput, Tfilter, Tbias, Toutput>* Get( in Get()
325 MklConvFwdPrimitive<Tinput, Tfilter, Tbias, Toutput>* conv_fwd = nullptr; in Get()
330 new MklConvFwdPrimitive<Tinput, Tfilter, Tbias, Toutput>(convFwdDims); in Get()
334 dynamic_cast<MklConvFwdPrimitive<Tinput, Tfilter, Tbias, Toutput>*>( in Get()
339 conv_fwd = new MklConvFwdPrimitive<Tinput, Tfilter, Tbias, Toutput>( in Get()
647 MklConvFwdPrimitive<Tinput, Tfilter, Tbias, Ttemp_output>* conv_fwd = in Compute()
|