Searched refs:ArgMaxBuilder (Results 1 – 3 of 3) sorted by relevance
26 ArgMaxBuilder::ArgMaxBuilder() {} in ArgMaxBuilder() function in OHOS::NeuralNetworkRuntime::Ops::ArgMaxBuilder28 ArgMaxBuilder::~ArgMaxBuilder() {} in ~ArgMaxBuilder()30 OH_NN_ReturnCode ArgMaxBuilder::SetAxis(const std::shared_ptr<NNTensor>& tensor) in SetAxis()49 OH_NN_ReturnCode ArgMaxBuilder::SetTopK(const std::shared_ptr<NNTensor>& tensor) in SetTopK()68 OH_NN_ReturnCode ArgMaxBuilder::SetKeepdims(const std::shared_ptr<NNTensor>& tensor) in SetKeepdims()87 OH_NN_ReturnCode ArgMaxBuilder::SetOutMaxValue(const std::shared_ptr<NNTensor>& tensor) in SetOutMaxValue()112 OH_NN_ReturnCode ArgMaxBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build()158 LiteGraphPrimitvePtr ArgMaxBuilder::GetPrimitive() in GetPrimitive()169 REGISTER_OPS(ArgMaxBuilder, OH_NN_OPS_ARG_MAX);
27 class ArgMaxBuilder : public OpsBuilder {29 typedef OH_NN_ReturnCode (ArgMaxBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);31 ArgMaxBuilder();32 ~ArgMaxBuilder() override;51 {OH_NN_ARG_MAX_AXIS, &ArgMaxBuilder::SetAxis},52 {OH_NN_ARG_MAX_KEEPDIMS, &ArgMaxBuilder::SetKeepdims},53 {OH_NN_ARG_MAX_TOP_K, &ArgMaxBuilder::SetTopK},54 {OH_NN_ARG_MAX_OUT_MAX_VALUE, &ArgMaxBuilder::SetOutMaxValue}
42 ArgMaxBuilder m_builder;