/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/tests/unittests/ |
H A D | GraphTest.cpp | 136 auto *top = F->createTopK("top", add, 5); in TEST() 1325 auto *topK = F->createTopK("topK", input, 12); in TEST() 1555 TopKNode *topK = F->createTopK("add", input, 5); in TEST() 1954 TopKNode *TKN = F->createTopK("topk", I, 3); in TEST() 2073 TopKNode *TKN = F->createTopK("topk", I, 3); in TEST() 2097 TopKNode *TKN2 = F2->createTopK("topk", I, 3); in TEST() 2145 F2->createTopK("topk", I, 3); in TEST()
|
H A D | OnnxExporterTest.cpp | 886 TopKNode *TK = F_->createTopK("topk", convertW, 5); in TEST_F()
|
H A D | PartitionerTest.cpp | 1153 auto *R1 = F_->createTopK("TopK1", inp1, 2); in TEST_F() 1154 auto *R2 = F_->createTopK("TopK2", inp2, 2); in TEST_F()
|
H A D | TypeAToTypeBFunctionConverterTest.cpp | 524 auto *topK = F->createTopK("topK", input, 3); in TEST()
|
H A D | QuantizationTest.cpp | 1540 Node *downsample = F->createTopK("gru.downsample", hidden, embeddingSize / 2); in createGRUForQuantization() 2886 auto *TK = F->createTopK("TopK", input, 3); in TEST()
|
H A D | GraphOptzTest.cpp | 1887 auto *topK = F_->createTopK("topK", input, /*k=*/5); in TEST_F() 4260 TopKNode *TK = F_->createTopK("topk", convertW, 5); in TEST_F() 4377 auto *topK = F_->createTopK("topK", mul3, 2); in TEST_F()
|
H A D | OperatorTest.cpp | 3318 auto *R = F_->createTopK("TopK", inp, 3); in TEST_P() 3686 auto *R1 = F_->createTopK("TopK1", inp1, 2); in TEST_P() 3687 auto *R2 = F_->createTopK("TopK2", inp2, 2); in TEST_P() 3795 auto *R = F_->createTopK("TopK", inp, 1, topKElemKind); in topK1Template() 3839 auto *TK = F_->createTopK("TopK", INV, 3); in TEST_P()
|
/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/examples/ |
H A D | fr2en.cpp | 344 auto *topK = F_->createTopK("decoder.topK", FC, 1); in loadDecoder()
|
/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/include/glow/Graph/ |
H A D | Graph.h | 1360 TopKNode *createTopK(llvm::StringRef name, NodeValue input, unsigned_t k); 1362 TopKNode *createTopK(llvm::StringRef name, NodeValue input, unsigned_t k,
|
/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/include/glow/Importer/ |
H A D | CommonOperatorLoader.h | 796 auto *R = G_->createTopK(opName, in, k); in loadTopK()
|
/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/lib/Graph/ |
H A D | Graph.cpp | 2410 TopKNode *Function::createTopK(llvm::StringRef name, NodeValue input, in createTopK() function in Function 2422 TopKNode *Function::createTopK(llvm::StringRef name, NodeValue input, in createTopK() function in Function 2424 return createTopK(name, input, k, ElemKind::Int64ITy); in createTopK()
|
/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/torch_glow/src/ |
H A D | PyTorchModelLoader.cpp | 3588 auto *glowNode = F_.createTopK("TopK", input, (unsigned_t)k); in loadTopK()
|