Searched refs:createFlatten (Results 1 – 8 of 8) sorted by relevance
/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/include/glow/Importer/ |
H A D | CommonOperatorLoader.h | 400 auto *FN = G_->createFlatten("reshapeInput", in, axis); in loadSoftmax() 731 auto *node = G_->createFlatten(opName, in, axis); in loadFlatten()
|
/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/include/glow/Graph/ |
H A D | Graph.h | 871 ReshapeNode *createFlatten(llvm::StringRef name, NodeValue input,
|
/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/lib/Optimizer/Lower/ |
H A D | Lower.cpp | 272 Node *x2 = F->createFlatten(DECORATE_NODE_NAME(FCG, "output", "flatten"), in lowerFullyConnectedGradNode()
|
/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/tests/unittests/ |
H A D | HabanaGlowTest.cpp | 1175 auto flatten = F_->createFlatten("flatten", input, /*axis=*/2); in TEST_F()
|
H A D | OperatorTest.cpp | 882 fltn2 = F->createFlatten("flatten", gthB, 2); in testNonMaxSuppressionWithGather() 13544 NodeValue reshape4Dto2DAxis1 = F->createFlatten("flat4Dto2Da1", tensor4D, 1); in testFlatten() 13549 NodeValue reshape4Dto2DAxis2 = F->createFlatten("flat4Dto2Da2", tensor4D, 2); in testFlatten() 13554 NodeValue reshape4Dto2DAxis3 = F->createFlatten("flat4Dto2Da3", tensor4D, 3); in testFlatten() 13563 NodeValue reshape4Dto2DAxis4 = F->createFlatten("flat4Dto2Da4", tensor4D, 4); in testFlatten() 13576 NodeValue reshape1Dto2DAxis1 = F->createFlatten("flat1Dto2D", tensor1D, 1); in testFlatten()
|
/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/lib/Graph/ |
H A D | Graph.cpp | 1076 input = createFlatten(name.str() + ".reshape2D", input, axis); in createFullyConnected() 1571 ReshapeNode *Function::createFlatten(llvm::StringRef name, NodeValue input, in createFlatten() function in Function 3130 input = createFlatten(name.str() + ".reshape2D", input, axis); in createFullyConnected()
|
/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/torch_glow/src/ |
H A D | PyTorchModelLoader.cpp | 3472 auto *FN = F_.createFlatten("reshapeInput", in, dim); in loadSoftMax()
|
/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/lib/Importer/ |
H A D | ONNXModelLoader.cpp | 2202 in = G_->createFlatten("fc.in", in, axis); in loadFCTransposed()
|