Home
last modified time | relevance | path

Searched refs:OpType_CosineSimilarity (Results 1 – 6 of 6) sorted by relevance

/dports/misc/mnn/MNN-1.2.0/tools/converter/source/onnx/
H A DCosineSimilarityOnnx.cpp14 return MNN::OpType_CosineSimilarity; in opType()
/dports/misc/mnn/MNN-1.2.0/source/shape/
H A DShapeCosineSimilarity.cpp49 REGISTER_SHAPE_INPUTS(CosineSimilaritySize, OpType_CosineSimilarity, (std::vector<int>{2}));
/dports/misc/mnn/MNN-1.2.0/backupcode/cpubackend/
H A DCPUCosineSimilarity.cpp88 REGISTER_CPU_OP_CREATOR(CPUCosineSimilarityCreator, OpType_CosineSimilarity);
/dports/misc/mnn/MNN-1.2.0/source/geometry/
H A DGeometryCosineSimilarity.cpp237 GeometryComputer::registerGeometryComputer(comp, {OpType_CosineSimilarity}); in _create()
/dports/misc/mnn/MNN-1.2.0/express/
H A DNeuralNetWorkOp.cpp1644 cosineSimilarityOp->type = MNN::OpType_CosineSimilarity; in _CosineSimilarity()
/dports/misc/mnn/MNN-1.2.0/schema/current/
H A DMNN_generated.h197 OpType_CosineSimilarity = 108, enumerator
364 OpType_CosineSimilarity, in EnumValuesOpType()