Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dmkl_util.h759 inline size_t GetSerializeBufferSize() const { return sizeof(MklShapeData); } in GetSerializeBufferSize() function
762 CHECK(buf_size >= GetSerializeBufferSize()) in SerializeMklDnnShape()
774 CHECK(buf_size >= GetSerializeBufferSize()) in DeSerializeMklDnnShape()
983 second_shape.AddDim(mkl_shape.GetSerializeBufferSize()); in AllocateOutputSetMklShape()
1023 second_shape.AddDim(mkl_shape.GetSerializeBufferSize()); in AllocateOutputSetMklShape()
/external/tensorflow/tensorflow/core/kernels/
Dmkl_conv_ops_test.cc111 auto size = static_cast<int64>(non_mkl_shape.GetSerializeBufferSize()); in NonMklTensor()
Dmkl_concat_op.cc477 tf_shape_output.AddDim(dnn_shape_output.GetSerializeBufferSize()); in CallEigenVersion()