Searched refs:mklshape (Results 1 – 1 of 1) sorted by relevance
876 inline void GetMklShape(OpKernelContext* ctext, int n, MklShape* mklshape) { in GetMklShape() argument877 mklshape->DeSerializeMklShape( in GetMklShape()887 inline void GetMklShape(OpKernelContext* ctext, int n, MklDnnShape* mklshape) { in GetMklShape() argument888 mklshape->DeSerializeMklDnnShape( in GetMklShape()