Searched refs:GetFilterMemoryFormat (Results 1 – 2 of 2) sorted by relevance
116 memory::format GetFilterMemoryFormat() const { return context_.filter_fmt; } in GetFilterMemoryFormat() function in tensorflow::MklConvBwdInputPrimitive441 conv_bwd_input->GetFilterMemoryFormat()) { in Compute()
163 memory::format GetFilterMemoryFormat() const { return context_.filter_fmt; } in GetFilterMemoryFormat() function in tensorflow::MklConvFwdPrimitive1111 if (filter_md.data.format != conv_fwd->GetFilterMemoryFormat()) { in Compute()1124 GetCachedFilter(context, conv_fwd->GetFilterMemoryFormat()); in Compute()