MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / SerializeMklDnnShape

Method SerializeMklDnnShape

tensorflow/core/util/mkl_util.h:582–586  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

580 inline size_t GetSerializeBufferSize() const { return sizeof(MklShapeData); }
581
582 void SerializeMklDnnShape(unsigned char* buf, size_t buf_size) const {
583 CHECK(buf_size >= GetSerializeBufferSize())
584 << "Buffer size is too small to SerializeMklDnnShape";
585 *reinterpret_cast<MklShapeData*>(buf) = data_;
586 }
587
588 void DeSerializeMklDnnShape(const unsigned char* buf, size_t buf_size) {
589 // Make sure buffer holds at least is_mkl_tensor_.

Callers 2

GetMklMetaTensorFunction · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected