Searched refs:MklConvFwdParams (Results 1 – 1 of 1) sorted by relevance
66 struct MklConvFwdParams { struct82 MklConvFwdParams(memory::dims src_dims, memory::dims filter_dims, in MklConvFwdParams() argument104 explicit MklConvFwdPrimitive(const MklConvFwdParams& convFwdDims) in MklConvFwdPrimitive()214 void Setup(const MklConvFwdParams& convFwdDims) { in Setup()324 const MklConvFwdParams& convFwdDims, bool do_not_cache) { in Get()361 static string CreateKey(const MklConvFwdParams& convFwdDims) { in CreateKey()399 MklPrimitive* GetConvFwd(const MklConvFwdParams& convFwdDims) { in GetConvFwd()404 void SetConvFwd(const MklConvFwdParams& convFwdDims, MklPrimitive* op) { in SetConvFwd()1072 MklConvFwdParams convFwdDims( in Compute()1226 MklConvFwdParams& params) { in ExtendConvFwdParams()[all …]