Searched refs:tfLayout_ (Results 1 – 1 of 1) sorted by relevance
162 if (tfLayout_) CHECK_EQ(dnnLayoutDelete_F32(tfLayout_), E_SUCCESS); in ~MklShape()192 CHECK_EQ(dnnLayoutCreate_F32(&tfLayout_, dimension, sizes, strides), in SetTfLayout()237 const dnnLayout_t GetTfLayout() const { return tfLayout_; } in GetTfLayout()239 return isMklTensor_ ? mklLayout_ : tfLayout_; in GetCurLayout()290 curLayout = tfLayout_; in GetConvertedFlatData()369 CHECK_EQ(dnnLayoutDeserialize_F32(&tfLayout_, in DeSerializeMklShape()395 dnnLayoutSerialize_F32(tfLayout_, buf + TF_LAYOUT_OFFSET(dimension_)), in SerializeMklShape()404 dnnLayout_t tfLayout_ = nullptr; // Pointer to layout of corresponding variable