Searched refs:MklConvFwdParams (Results 1 – 1 of 1) sorted by relevance
58 struct MklConvFwdParams { struct78 MklConvFwdParams(memory::dims src_dims, memory::dims filter_dims, in MklConvFwdParams() function100 explicit MklConvFwdPrimitive(const MklConvFwdParams& convFwdDims) in MklConvFwdPrimitive()209 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()401 MklPrimitive* GetConvFwd(const MklConvFwdParams& convFwdDims) { in GetConvFwd()406 void SetConvFwd(const MklConvFwdParams& convFwdDims, MklPrimitive* op) { in SetConvFwd()653 MklConvFwdParams convFwdDims( in Compute()818 MklConvFwdParams& params) { in ExtendConvFwdParams()[all …]