Searched refs:MklConvFwdPrimitiveFactory (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_conv_ops.cc | 321 class MklConvFwdPrimitiveFactory : public MklPrimitiveFactory<T> { class
335 MklConvFwdPrimitiveFactory<T, Tinput, Tfilter, Tbias, in Get()
341 MklConvFwdPrimitiveFactory<T, Tinput, Tfilter, Tbias, in Get()
351 MklConvFwdPrimitiveFactory() {} in MklConvFwdPrimitiveFactory() function in tensorflow::MklConvFwdPrimitiveFactory
352 ~MklConvFwdPrimitiveFactory() {} in ~MklConvFwdPrimitiveFactory()
356 static MklConvFwdPrimitiveFactory& GetInstance() { in GetInstance()
357 static MklConvFwdPrimitiveFactory instance_; in GetInstance()
1079 conv_fwd = MklConvFwdPrimitiveFactory<float, Tinput, Tfilter, Tbias, in Compute()
|