Lines Matching refs:GetTensorLayoutDim
430 int32_t *GetTensorLayoutDim(uint32_t dim) in GetTensorLayoutDim() function
1945 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()
3755 … index = index * GetTensorLayoutDim(dim)[k] + loadCoord[k]; in iterate()
4462 for (int32_t i0 = 0; i0 < GetTensorLayoutDim(dim)[0]; ++i0) in iterate()
4464 for (int32_t i1 = 0; i1 < GetTensorLayoutDim(dim)[1]; ++i1) in iterate()
4466 for (int32_t i2 = 0; i2 < GetTensorLayoutDim(dim)[2]; ++i2) in iterate()
4468 for (int32_t i3 = 0; i3 < GetTensorLayoutDim(dim)[3]; ++i3) in iterate()
4470 for (int32_t i4 = 0; i4 < GetTensorLayoutDim(dim)[4]; ++i4) in iterate()
4476 index = index * GetTensorLayoutDim(dim)[k] + tensorCoord[k]; in iterate()
4507 … if (loadCoord[k] < 0 || loadCoord[k] >= GetTensorLayoutDim(dim)[k]) in iterate()
4521 … index = index * GetTensorLayoutDim(dim)[k] + loadCoord[k]; in iterate()