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