Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_pooling_ops_common.h594 size_t GetNumTElements(const memory::desc& pd) { in GetNumTElements() function
673 output_tf_shape.AddDim(this->GetNumTElements(dst_pd)); in AllocateOutputTensor()
726 output_tf_shape.AddDim(this->GetNumTElements(dst_pd)); in AllocateOutputTensor()