Searched refs:GetTensorLayoutDim (Results 1 – 1 of 1) sorted by relevance
430 int32_t *GetTensorLayoutDim(uint32_t dim) in GetTensorLayoutDim() function1945 css << ", " << GetTensorLayoutDim(dim)[j]; in initProgramsGLSL()1961 css << ", " << GetTensorLayoutDim(dim)[j]; in initProgramsGLSL()3696 for (int32_t i0 = 0; i0 < GetTensorLayoutDim(dim)[0]; ++i0) in iterate()3698 for (int32_t i1 = 0; i1 < GetTensorLayoutDim(dim)[1]; ++i1) in iterate()3700 for (int32_t i2 = 0; i2 < GetTensorLayoutDim(dim)[2]; ++i2) in iterate()3702 for (int32_t i3 = 0; i3 < GetTensorLayoutDim(dim)[3]; ++i3) in iterate()3704 for (int32_t i4 = 0; i4 < GetTensorLayoutDim(dim)[4]; ++i4) in iterate()3710 index = index * GetTensorLayoutDim(dim)[k] + tensorCoord[k]; in iterate()3741 … if (loadCoord[k] < 0 || loadCoord[k] >= GetTensorLayoutDim(dim)[k]) in iterate()[all …]