| /dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/lib/Backends/CPU/ClassGen/ |
| H A D | CPUSpecificNodesVerification.h | 25 ShapeNHWC idim(getInput().getType()->dims()); in verify()
26 ShapeNHWC odim(getResult().getType()->dims()); in verify()
29 ShapeNHWC exp(idim.n, outSz.first, outSz.second, getBias().dims()[0]); in verify()
|
| /dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/lib/Backends/OpenCL/ |
| H A D | kernels.cl | 27 /// ShapeNHWC in Type.h
33 } ShapeNHWC;
766 ShapeNHWC ldim, ShapeNHWC rdim) {
816 ShapeNHWC ddim, ShapeNHWC ldim, ShapeNHWC rdim,
1123 ShapeNHWC odim, ShapeNHWC idim,
1599 ShapeNHWC odim, ShapeNHWC idim, ShapeNHWC shuffle) { \
1617 ShapeNHWC odim, ShapeNHWC idim, ShapeNHWC shuffle) { \
1633 ShapeNHWC odim, ShapeNHWC idim, ShapeNHWC offset, \
1660 ShapeNHWC odim, ShapeNHWC idim, ShapeNHWC offset, \
1674 ShapeNHWC odim, ShapeNHWC idim, ShapeNHWC offset) { \
[all …]
|
| H A D | OpenCL.cpp | 745 return ShapeNHWC(llvm::ArrayRef<dim_t>(ones)); in shapeFromDims()
969 ShapeNHWC offset = shapeFromDims(ET->getOffsets()); in execute()
990 ShapeNHWC offset = shapeFromDims(IT->getOffsets()); in execute()
1131 ShapeNHWC dim(LRN->getDest()->getType()->dims()); in execute()
1151 ShapeNHWC dim(LRN->getDest()->getType()->dims()); in execute()
1263 auto destGradDim = ShapeNHWC(destGrad->dims()); in execute()
1264 auto srcDim = ShapeNHWC(src->dims()); in execute()
1265 auto filterGradDim = ShapeNHWC(filterGrad->dims()); in execute()
1319 ShapeNHWC odim(PM->getDest()->getType()->dims()); in execute()
1320 ShapeNHWC idim(PM->getSrc()->getType()->dims()); in execute()
[all …]
|
| /dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/tests/unittests/ |
| H A D | NodeSplittingTest.cpp | 208 {ShapeNHWC::DimN, ShapeNHWC::DimH}, {2, 2}); in TEST_F()
215 {ShapeNHWC::DimN, ShapeNHWC::DimH, ShapeNHWC::DimW}, in TEST_F()
224 {ShapeNHWC::DimN, ShapeNHWC::DimH, ShapeNHWC::DimW, ShapeNHWC::DimC}, in TEST_F()
287 {ShapeNHWC::DimH, ShapeNHWC::DimW}, {3, 3}); in TEST_F()
317 auto splitOption = SplitNodeByNumChunks({ShapeNHWC::DimC}, {numChunks}); in splitConv2DGrouped()
363 std::vector<size_t> splitDims = {ShapeNHWC::DimH, ShapeNHWC::DimW}; in TEST_F()
612 {ShapeNHWC::DimH, ShapeNHWC::DimW}, {2, 2}); in TEST_F()
622 std::vector<size_t> splitDims = {ShapeNHWC::DimH, ShapeNHWC::DimW}; in TEST_F()
712 auto splitOption = SplitNodeByNumChunks({ShapeNHWC::DimH}, {3}); in TEST_F()
898 {ShapeNHWC::DimH, ShapeNHWC::DimW}, {2, 2}); in TEST_F()
[all …]
|
| /dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/include/glow/Backends/ |
| H A D | LayoutConverter.h | 57 auto dimsNHWC = ShapeNHWC(CN->getResult().getType()->dims()); in convertConvToNCHWConv()
79 auto dimsNHWC = ShapeNHWC(PN->getResult().getType()->dims()); in convertMaxPoolToNCHWPool()
123 auto dimsNHWC = ShapeNHWC(PN->getResult().getType()->dims()); in convertAvgPoolToNCHWPool()
|
| /dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/lib/Backends/Habana/ClassGen/ |
| H A D | HabanaSpecificNodesVerification.h | 40 ShapeNHWC idim(src.getType()->dims()); in verifyConvolution()
41 ShapeNHWC odim(dest.getType()->dims()); in verifyConvolution()
|
| /dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/lib/Optimizer/GraphOptimizer/ |
| H A D | TrainingPreparation.cpp | 31 ShapeNHWC idim = ShapeNHWC(CN->getInput().dims()); in defaultTensorInitializer()
|
| H A D | NodeSplitting.cpp | 1397 getConv2DInputIdxAndMaps<ShapeNHWC>(dyn_cast<ConvolutionNode>(node)), in splitNode()
1398 {}, Conv2DSplitNodeModifier<ShapeNHWC>); in splitNode()
1411 getPoolInputIdxAndMaps<MaxPoolNode, ShapeNHWC>( in splitNode()
1414 PoolSplitNodeModifier<MaxPoolNode, ShapeNHWC>); in splitNode()
1420 getPoolInputIdxAndMaps<AvgPoolNode, ShapeNHWC>( in splitNode()
1422 {}, PoolSplitNodeModifier<AvgPoolNode, ShapeNHWC>); in splitNode()
|
| /dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/lib/IR/ |
| H A D | IRBuilder.cpp | 66 ShapeNHWC idim = ShapeNHWC(input->dims()); in createMaxPoolWithArgmaxOp()
117 ShapeNHWC idim(input->dims()); in createAvgPoolOp()
|
| /dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/include/glow/Base/ |
| H A D | Type.h | 66 struct ShapeNHWC { struct
80 template <typename T> explicit ShapeNHWC(llvm::ArrayRef<T> shape) { in ShapeNHWC() argument
88 ShapeNHWC(dim_t samples, dim_t height, dim_t width, dim_t channels) in ShapeNHWC() function
91 bool equals(const ShapeNHWC &other) const { in equals() argument
359 inline bool operator==(const ShapeNHWC &LHS, const ShapeNHWC &RHS) {
|
| /dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/lib/Importer/ |
| H A D | TFLiteModelLoader.cpp | 1358 ShapeNHWC inputShape = ShapeNHWC(input.dims()); in loadPool2D()
1359 ShapeNHWC outputShape = ShapeNHWC(outTy->dims()); in loadPool2D()
1455 ShapeNHWC inputShape = ShapeNHWC(input.dims()); in loadConv2D()
1456 ShapeNHWC filterShape = ShapeNHWC(filter.dims()); in loadConv2D()
1457 ShapeNHWC outputShape = ShapeNHWC(outTy->dims()); in loadConv2D()
1547 ShapeNHWC inputShape = ShapeNHWC(input.dims()); in loadDepthwiseConv2D()
1548 ShapeNHWC filterShape = ShapeNHWC(filter.dims()); in loadDepthwiseConv2D()
1549 ShapeNHWC outputShape = ShapeNHWC(outTy->dims()); in loadDepthwiseConv2D()
|
| H A D | Caffe2ModelLoader.cpp | 347 ShapeNHWC idim = ShapeNHWC(finalInType->dims()); in loadConv()
445 ShapeNHWC idim = ShapeNHWC(finalInType->dims()); in loadConvQuantized()
663 ShapeNHWC idim = ShapeNHWC(finalInType->dims()); in loadConvTranspose()
850 ShapeNHWC idim = ShapeNHWC(finalInType->dims()); in loadOperator()
|
| H A D | ONNXModelLoader.cpp | 1314 ShapeNHWC idim = ShapeNHWC(tr->getResult().dims()); in loadConv1D()
1431 ShapeNHWC idim = ShapeNHWC(tr->getResult().dims()); in loadConv()
1483 ShapeNHWC idim(input.dims()); in loadTensorwiseQuantizedConvolution()
1532 ShapeNHWC idim(input.dims()); in loadChannelwiseQuantizedConvolution()
1636 ShapeNHWC idim = ShapeNHWC(tr->getResult().dims()); in loadConvTranspose()
3710 ShapeNHWC idim(input.dims()); in loadAdaptiveAvgPool()
|
| /dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/lib/Graph/ |
| H A D | Nodes.cpp | 115 const ShapeNHWC &idim, const ShapeNHWC &odim, in verifyConvFilter()
267 ShapeNHWC idim(src.getType()->dims()); in verifyConvTranspose()
268 ShapeNHWC odim(dest.getType()->dims()); in verifyConvTranspose()
587 return verifyConvolution<ShapeNHWC>(getInput(), getResult(), getFilter(), in verify()
607 isValid = verifyConvolution<ShapeNHWC>( in verify()
692 isValid &= verifyConvolution<ShapeNHWC>( in verify()
764 return verifyPool<ShapeNHWC>(getInput(), getResult(), Kernels_, Strides_, in verify()
801 ShapeNHWC idim(getInput().getType()->dims()); in verify()
802 ShapeNHWC odim(getOriginalOutputForResult().getType()->dims()); in verify()
863 isValid &= verifyPool<ShapeNHWC>( in verify()
[all …]
|
| H A D | VerifierHelper.cpp | 27 void glow::reportContext(const ShapeNHWC &shapeNHWC) { in reportContext()
|
| H A D | Graph.cpp | 714 ShapeNHWC idim = ShapeNHWC(input.dims()); in assertConvTransposeDims()
724 ShapeNHWC filterDims(filter.dims()); in assertConvTransposeDims()
740 ShapeNHWC idim = ShapeNHWC(input.dims()); in assertConvDims()
750 ShapeNHWC filterDims(filter.dims()); in assertConvDims()
905 ShapeNHWC idim = ShapeNHWC(input.dims()); in createMaxPool()
936 ShapeNHWC idim = ShapeNHWC(input.dims()); in createAvgPool()
968 ShapeNHWC idim = ShapeNHWC(input.dims()); in createAvgPool()
2776 ShapeNHWC idim = ShapeNHWC(input.dims()); in createConv()
3057 ShapeNHWC idim = ShapeNHWC(input.dims()); in createConvTranspose()
5601 auto filterDims = ShapeNHWC(node->getFilter().getType()->dims()); in isConvolutionSameAsFullyConnected()
[all …]
|
| /dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/lib/Backends/Interpreter/ |
| H A D | InterpreterNodes.cpp | 226 ShapeNHWC odim(outW.dims()); in fwdConvolutionInstFloatImpl()
227 ShapeNHWC idim(inW.dims()); in fwdConvolutionInstFloatImpl()
293 ShapeNHWC odim(outW.dims()); in fwdConvolutionInstQuantizedImpl()
294 ShapeNHWC idim(inW.dims()); in fwdConvolutionInstQuantizedImpl()
391 ShapeNHWC odim(outW.dims()); in fwdConvTransposeInstFloatImpl()
392 ShapeNHWC idim(inW.dims()); in fwdConvTransposeInstFloatImpl()
511 ShapeNHWC odim(outG.dims()); in fwdConvolutionGradInst()
512 ShapeNHWC idim(inW.dims()); in fwdConvolutionGradInst()
798 ShapeNHWC idim(inW.dims()); in fwdChannelwiseQuantizedConv2DInstImpl()
1459 ShapeNHWC idim(inG.dims()); in fwdAdaptiveAvgPoolGradInst()
[all …]
|
| /dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/include/glow/Graph/ |
| H A D | VerifierHelper.h | 58 void reportContext(const ShapeNHWC &shapeNHWC);
|
| /dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/lib/Optimizer/Lower/ |
| H A D | Lower.cpp | 894 auto inpDims = ShapeNHWC(input.getType()->dims()); in lowerConvolutionToFullyConnected()
900 auto filterDims = ShapeNHWC(filter.getType()->dims()); in lowerConvolutionToFullyConnected()
908 auto outDims = ShapeNHWC(output.getType()->dims()); in lowerConvolutionToFullyConnected()
913 CN.getName().str(), input, weights, bias, outTyFC, ShapeNHWC::DimC); in lowerConvolutionToFullyConnected()
939 ShapeNHWC idim = ShapeNHWC(in.dims()); in lowerGroupConvolutionNode()
|
| /dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/torch_glow/src/ |
| H A D | PyTorchModelLoader.cpp | 1196 glow::ShapeNHWC inputShape(input.dims()); in loadQuantizedConvImpl()
1197 glow::ShapeNHWC weightShape(weight.dims()); in loadQuantizedConvImpl()
2292 glow::ShapeNHWC weightsShape(weights.dims()); in loadConvolution()
2307 glow::ShapeNHWC inputShape(input.dims()); in loadConvolution()
2385 glow::ShapeNHWC weightsShape(weights.dims()); in loadConv2D()
2390 glow::ShapeNHWC inputShape(input.dims()); in loadConv2D()
2731 glow::ShapeNHWC inputShape(input.dims()); in loadQuantizedConvUnpackedImpl()
2732 glow::ShapeNHWC weightShape(weights.dims()); in loadQuantizedConvUnpackedImpl()
2996 auto idim = glow::ShapeNHWC(input.dims()); in loadAdaptiveAvgPool2d()
|
| /dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/lib/Exporter/ |
| H A D | ONNXModelWriter.cpp | 1717 const auto outShape = ShapeNHWC(node->getResult().dims()); in writeAdaptiveAvgPool()
|