Searched refs:GetDimension3D (Results 1 – 2 of 2) sorted by relevance
281 depth = mklInputShape->GetDimension3D('C'); in Init()282 tensor_in_cols = mklInputShape->GetDimension3D('W'); in Init()283 tensor_in_rows = mklInputShape->GetDimension3D('H'); in Init()284 tensor_in_planes = mklInputShape->GetDimension3D('D'); in Init()285 tensor_in_batch = mklInputShape->GetDimension3D('N'); in Init()
519 inline size_t GetDimension3D(char dimension) const { in GetDimension3D() function