Searched defs:MklBatchMatMulHelper (Results 1 – 1 of 1) sorted by relevance
31 struct MklBatchMatMulHelper { struct32 using dims = dnnl::memory::dims;37 void ExpandInputDimsToOutputShape(const TensorShape& input_shape, in ExpandInputDimsToOutputShape()51 std::unique_ptr<MklMatMulParams> CreateMatMulParams( in CreateMatMulParams()