Searched refs:GetDimension3D (Results 1 – 2 of 2) sorted by relevance
266 depth = mklInputShape->GetDimension3D('C'); in Init()267 tensor_in_cols = mklInputShape->GetDimension3D('W'); in Init()268 tensor_in_rows = mklInputShape->GetDimension3D('H'); in Init()269 tensor_in_planes = mklInputShape->GetDimension3D('D'); in Init()270 tensor_in_batch = mklInputShape->GetDimension3D('N'); in Init()
328 inline size_t GetDimension3D(char dimension) const { in GetDimension3D() function