Searched refs:mklLayout_ (Results 1 – 1 of 1) sorted by relevance
161 if (mklLayout_) CHECK_EQ(dnnLayoutDelete_F32(mklLayout_), E_SUCCESS); in ~MklShape()172 void SetMklLayout(dnnLayout_t mklLayout) { mklLayout_ = mklLayout; } in SetMklLayout()176 dnnLayoutCreateFromPrimitive_F32(&mklLayout_, (dnnPrimitive_t)primitive, in SetMklLayout()236 const dnnLayout_t GetMklLayout() const { return mklLayout_; } in GetMklLayout()239 return isMklTensor_ ? mklLayout_ : tfLayout_; in GetCurLayout()288 curLayout = mklLayout_; in GetConvertedFlatData()366 CHECK_EQ(dnnLayoutDeserialize_F32(&mklLayout_, in DeSerializeMklShape()391 CHECK_EQ(dnnLayoutSerialize_F32(mklLayout_, in SerializeMklShape()403 dnnLayout_t mklLayout_ = nullptr; // Pointer to the MKL layout variable