Searched refs:GetNumTElements (Results 1 – 1 of 1) sorted by relevance
594 size_t GetNumTElements(const memory::desc& pd) { in GetNumTElements() function673 output_tf_shape.AddDim(this->GetNumTElements(dst_pd)); in AllocateOutputTensor()726 output_tf_shape.AddDim(this->GetNumTElements(dst_pd)); in AllocateOutputTensor()