Home
last modified time | relevance | path

Searched refs:RangeBuilder (Results 1 – 3 of 3) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Drange_builder.cpp27 RangeBuilder::RangeBuilder() {} in RangeBuilder() function in OHOS::NeuralNetworkRuntime::Ops::RangeBuilder
29 RangeBuilder::~RangeBuilder() {} in ~RangeBuilder()
31 OH_NN_ReturnCode RangeBuilder::SetStart(const std::shared_ptr<NNTensor>& tensor) in SetStart()
53 OH_NN_ReturnCode RangeBuilder::SetLimit(const std::shared_ptr<NNTensor>& tensor) in SetLimit()
75 OH_NN_ReturnCode RangeBuilder::SetDelta(const std::shared_ptr<NNTensor>& tensor) in SetDelta()
97 OH_NN_ReturnCode RangeBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build()
143 LiteGraphPrimitvePtr RangeBuilder::GetPrimitive() in GetPrimitive()
156 REGISTER_OPS(RangeBuilder, OH_NN_OPS_RANGE);
H A Drange_builder.h27 class RangeBuilder : public OpsBuilder {
29 typedef OH_NN_ReturnCode (RangeBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);
31 RangeBuilder();
32 ~RangeBuilder() override;
50 {OH_NN_RANGE_START, &RangeBuilder::SetStart},
51 {OH_NN_RANGE_LIMIT, &RangeBuilder::SetLimit},
52 {OH_NN_RANGE_DELTA, &RangeBuilder::SetDelta}
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Drange_test.cpp41 RangeBuilder m_builder;