Searched refs:convFwdDims (Results 1 – 1 of 1) sorted by relevance
100 explicit MklConvFwdPrimitive(const MklConvFwdParams& convFwdDims) in MklConvFwdPrimitive() argument104 Setup(convFwdDims); in MklConvFwdPrimitive()209 void Setup(const MklConvFwdParams& convFwdDims) { in Setup() argument211 if (convFwdDims.native_format) { in Setup()212 user_data_fmt = MklTensorFormatToMklDnnDataFormat(convFwdDims.tf_fmt); in Setup()218 {convFwdDims.src_dims}, MklDnnType<Tinput>(), user_data_fmt)); in Setup()220 context_.filter_md.reset(new memory::desc({convFwdDims.filter_dims}, in Setup()225 {convFwdDims.dst_dims}, MklDnnType<Toutput>(), user_data_fmt)); in Setup()227 if (!convFwdDims.bias_dims.empty()) in Setup()228 context_.bias_md.reset(new memory::desc({convFwdDims.bias_dims}, in Setup()[all …]