Home
last modified time | relevance | path

Searched refs:GetSerializeBufferSize (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_dequantize_op_test.cc76 {static_cast<int64>(mkl_shape.GetSerializeBufferSize())}); in CreateMklInput()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h580 inline size_t GetSerializeBufferSize() const { return sizeof(MklShapeData); } in GetSerializeBufferSize() function
583 CHECK(buf_size >= GetSerializeBufferSize()) in SerializeMklDnnShape()
595 CHECK(buf_size >= GetSerializeBufferSize()) in DeSerializeMklDnnShape()
745 second_shape.AddDim(mkl_shape.GetSerializeBufferSize()); in AllocateOutputSetMklShape()
767 second_shape.AddDim(mkl_shape.GetSerializeBufferSize());
951 auto size = static_cast<int64>(non_mkl_shape.GetSerializeBufferSize()); in GetMklMetaTensor()