Searched refs:inCperG (Results 1 – 6 of 6) sorted by relevance
/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/lib/LLVMIRCodeGen/libjit/ |
H A D | libjit_conv.cpp | 60 dim_t inCperG = inChannels / group; in libjit_quantized_conv2d_generic() local 110 for (size_t fd = 0; fd < inCperG; fd++) { in libjit_quantized_conv2d_generic() 162 dim_t inCperG = inChannels / group; in libjit_channelwise_quantized_conv2d_generic() local 216 for (dim_t fd = 0; fd < inCperG; fd++) { in libjit_channelwise_quantized_conv2d_generic() 251 dim_t inCperG = inChannels / group; in libjit_channelwise_quantized_conv3d_generic() local 352 dim_t inCperG = inChannels / group; in libjit_conv2d_f() local 381 for (dim_t cb = 0; cb < inCperG; cb += cbSize) { in libjit_conv2d_f() 577 dim_t inCperG = inWdims[3] / group; in libjit_conv_transpose_f() local 587 for (dim_t d = g * inCperG; d < (g + 1) * inCperG; d++) { in libjit_conv_transpose_f() 639 dim_t inCperG = inWdims[3] / group; in libjit_convolution_grad_f() local [all …]
|
/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/lib/Backends/OpenCL/ |
H A D | kernels_specialized_no_local_mem_conv.cl | 44 const dim_t inCperG = CONVK_IDIM_C / CONVK_GROUP; 46 const dim_t inChannelOffset = d / outCperG * inCperG; 71 for (dim_t fd = 0; fd < inCperG; fd++) {
|
H A D | kernels.cl | 1069 dim_t inCperG = idim.c / group; 1071 dim_t inChannelOffset = d / outCperG * inCperG; 1092 for (dim_t fd = 0; fd < inCperG; fd++) { 1142 dim_t inCperG = idim.c / group; 1144 dim_t inChannelOffset = d / outCperG * inCperG; 1167 for (dim_t fd = 0; fd < inCperG; fd++) { 1226 dim_t inCperG = inWdims.c / group; 1228 dim_t inChannelOffset = d / outCperG * inCperG; 1250 for (dim_t fd = 0; fd < inCperG; fd++) {
|
/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/lib/Backends/CPU/libjit_cpu/ |
H A D | libjit_cpu_conv.cpp | 265 dim_t inCperG = inChannels / group; in libjit_convDKKC8_f() local 290 eachPixelConv(n, d, numDepthRegs, depthStrips, sizeGroupY, inCperG, in libjit_convDKKC8_f()
|
/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/lib/Backends/Interpreter/ |
H A D | InterpreterNodes.cpp | 233 dim_t inCperG = idim.c / group; in fwdConvolutionInstFloatImpl() local 265 for (dim_t fd = 0; fd < inCperG; fd++) { in fwdConvolutionInstFloatImpl() 300 dim_t inCperG = idim.c / group; in fwdConvolutionInstQuantizedImpl() local 349 for (dim_t fd = 0; fd < inCperG; fd++) { in fwdConvolutionInstQuantizedImpl() 400 dim_t inCperG = idim.c / group; in fwdConvTransposeInstFloatImpl() local 421 for (dim_t d = g * inCperG; d < (g + 1) * inCperG; d++) { in fwdConvTransposeInstFloatImpl() 519 dim_t inCperG = idim.c / group; in fwdConvolutionGradInst() local 589 dim_t inCperG = idim.c / group; in fwdConvolution3DInstFloatImpl() local 661 dim_t inCperG = idim.c / group; in fwdConvolution3DInstQuantizedImpl() local 804 dim_t inCperG = idim.c / group; in fwdChannelwiseQuantizedConv2DInstImpl() local [all …]
|
/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/lib/Optimizer/Lower/ |
H A D | Lower.cpp | 941 unsigned inCperG = idim.c / group; in lowerGroupConvolutionNode() local 952 F->createSlice(BNG.getName(), in, {0, 0, 0, groupId * inCperG}, in lowerGroupConvolutionNode() 953 {idim.n, idim.h, idim.w, (groupId + 1) * inCperG}); in lowerGroupConvolutionNode() 956 {(groupId + 1) * outCperG, kdim.height, kdim.width, inCperG}); in lowerGroupConvolutionNode()
|