Searched refs:MklDnnMatMulOpBase (Results 1 – 3 of 3) sorted by relevance
31 class MklFusedMatMulOp : public MklDnnMatMulOpBase<T, T> {34 : MklDnnMatMulOpBase<T, T>(ctx) { in MklFusedMatMulOp()
115 class MklDnnQuantizedMatMulOp : public MklDnnMatMulOpBase<Tweight, Toutput> {140 : MklDnnMatMulOpBase<Tweight, Toutput>(context) { in MklDnnQuantizedMatMulOp()
382 class MklDnnMatMulOpBase : public OpKernel {384 explicit MklDnnMatMulOpBase(OpKernelConstruction* context) in MklDnnMatMulOpBase() function