Home
last modified time | relevance | path

Searched refs:NNTensor (Results 1 – 25 of 343) sorted by relevance

12345678910>>...14

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dnn_tensor.cpp37 NNTensor::~NNTensor() in ~NNTensor()
44 NNTensor::NNTensor(NNTensor&& tensor) noexcept in NNTensor() function in OHOS::NeuralNetworkRuntime::NNTensor
49 NNTensor& NNTensor::operator=(NNTensor&& tensor) noexcept in operator =()
331 void NNTensor::IdentifyOpParameter() in IdentifyOpParameter()
379 std::string NNTensor::GetName() const in GetName()
384 void* NNTensor::GetBuffer() const in GetBuffer()
467 bool NNTensor::IsDynamicShape() const in IsDynamicShape()
472 bool NNTensor::IsQuantTensor() const in IsQuantTensor()
477 bool NNTensor::IsScalar() const in IsScalar()
482 bool NNTensor::IsOpParameter() const in IsOpParameter()
[all …]
H A Dnn_tensor.h33 class NNTensor {
35 NNTensor() = default;
36 ~NNTensor();
37 NNTensor(NNTensor&& tensor) noexcept;
38 NNTensor& operator=(NNTensor&& tensor) noexcept;
40 NNTensor(const NNTensor& tensor) = delete;
41 NNTensor& operator=(const NNTensor& tensor) = delete;
80 bool CompareAttribute(const NNTensor& tensor) const;
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/inner_model/
H A Dnn_tensor_test.cpp42 NNTensor nnTensor;
60 NNTensor nnTensor;
74 NNTensor nnTensor;
88 NNTensor nnTensor;
102 NNTensor nnTensor;
119 NNTensor nnTensor;
138 NNTensor nnTensor;
156 NNTensor nnTensor;
174 NNTensor nnTensor;
192 NNTensor nnTensor;
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v2_0/inner_model/
H A Dnn_tensor_test.cpp42 NNTensor nnTensor;
60 NNTensor nnTensor;
74 NNTensor nnTensor;
88 NNTensor nnTensor;
102 NNTensor nnTensor;
119 NNTensor nnTensor;
138 NNTensor nnTensor;
156 NNTensor nnTensor;
174 NNTensor nnTensor;
192 NNTensor nnTensor;
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Dlstm_builder.h29 typedef OH_NN_ReturnCode (LSTMBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);
40 OH_NN_ReturnCode SetBidirectional(const std::shared_ptr<NNTensor>& tensor);
41 OH_NN_ReturnCode SetHasBias(const std::shared_ptr<NNTensor>& tensor);
42 OH_NN_ReturnCode SetInputSize(const std::shared_ptr<NNTensor>& tensor);
43 OH_NN_ReturnCode SetHiddenSize(const std::shared_ptr<NNTensor>& tensor);
44 OH_NN_ReturnCode SetNumLayers(const std::shared_ptr<NNTensor>& tensor);
45 OH_NN_ReturnCode SetNumDirections(const std::shared_ptr<NNTensor>& tensor);
46 OH_NN_ReturnCode SetDropout(const std::shared_ptr<NNTensor>& tensor);
47 OH_NN_ReturnCode SetZoneoutCell(const std::shared_ptr<NNTensor>& tensor);
48 OH_NN_ReturnCode SetZoneoutHidden(const std::shared_ptr<NNTensor>& tensor);
[all …]
H A Ddetection_post_process_builder.h27 typedef OH_NN_ReturnCode (DPPBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);
39 OH_NN_ReturnCode SetInputSize(const std::shared_ptr<NNTensor>& tensor);
40 OH_NN_ReturnCode SetScale(const std::shared_ptr<NNTensor>& tensor);
41 OH_NN_ReturnCode SetNmsIoUThreshold(const std::shared_ptr<NNTensor>& tensor);
42 OH_NN_ReturnCode SetNmsScoreThreshold(const std::shared_ptr<NNTensor>& tensor);
43 OH_NN_ReturnCode SetMaxDetections(const std::shared_ptr<NNTensor>& tensor);
44 OH_NN_ReturnCode SetDetectionsPerClass(const std::shared_ptr<NNTensor>& tensor);
45 OH_NN_ReturnCode SetMaxClassesPerDetection(const std::shared_ptr<NNTensor>& tensor);
46 OH_NN_ReturnCode SetNumClasses(const std::shared_ptr<NNTensor>& tensor);
47 OH_NN_ReturnCode SetUseRegularNms(const std::shared_ptr<NNTensor>& tensor);
[all …]
H A Dconv2d_builder.h28 typedef OH_NN_ReturnCode (Conv2DBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);
35 const std::vector<std::shared_ptr<NNTensor>>& allTensors) override;
42 const std::vector<std::shared_ptr<NNTensor>>& allTensors);
44 const std::vector<std::shared_ptr<NNTensor>>& allTensors);
46 const std::vector<std::shared_ptr<NNTensor>>& allTensors);
47 OH_NN_ReturnCode SetStrides(const std::shared_ptr<NNTensor>& tensor);
48 OH_NN_ReturnCode SetDilation(const std::shared_ptr<NNTensor>& tensor);
49 OH_NN_ReturnCode SetPad(const std::shared_ptr<NNTensor>& tensor);
50 OH_NN_ReturnCode SetGroup(const std::shared_ptr<NNTensor>& tensor);
51 OH_NN_ReturnCode SetActavitation(const std::shared_ptr<NNTensor>& tensor);
H A Dconv2d_transpose_builder.h28 typedef OH_NN_ReturnCode (Conv2DTransposeBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);
35 const std::vector<std::shared_ptr<NNTensor>>& allTensors) override;
42 const std::vector<std::shared_ptr<NNTensor>>& allTensors);
44 const std::vector<std::shared_ptr<NNTensor>>& allTensors);
45 OH_NN_ReturnCode SetStrides(const std::shared_ptr<NNTensor>& tensor);
46 OH_NN_ReturnCode SetDilation(const std::shared_ptr<NNTensor>& tensor);
47 OH_NN_ReturnCode SetPad(const std::shared_ptr<NNTensor>& tensor);
48 OH_NN_ReturnCode SetGroup(const std::shared_ptr<NNTensor>& tensor);
49 OH_NN_ReturnCode SetOutPadding(const std::shared_ptr<NNTensor>& tensor);
50 OH_NN_ReturnCode SetActivation(const std::shared_ptr<NNTensor>& tensor);
H A Ddepthwise_conv2d_native_builder.h28 …ypedef OH_NN_ReturnCode (DepthwiseConv2DNativeBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);
35 const std::vector<std::shared_ptr<NNTensor>>& allTensors) override;
40 …const std::vector<uint32_t>& outputsIndex, const std::vector<std::shared_ptr<NNTensor>>& allTensor…
41 OH_NN_ReturnCode SetIsPadMode(const std::shared_ptr<NNTensor>& tensor,
43 OH_NN_ReturnCode SetPadModeOrPaddings(const std::shared_ptr<NNTensor>& tensor);
45 const std::vector<std::shared_ptr<NNTensor>>& allTensors);
46 OH_NN_ReturnCode SetDilation(const std::shared_ptr<NNTensor>& tensor);
47 OH_NN_ReturnCode SetStrides(const std::shared_ptr<NNTensor>& tensor);
48 OH_NN_ReturnCode SetActivation(const std::shared_ptr<NNTensor>& tensor);
H A Dstrided_slice_builder.h27 typedef OH_NN_ReturnCode (StridedSliceBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);
34 const std::vector<std::shared_ptr<NNTensor>>& allTensors) override;
41 const std::vector<std::shared_ptr<NNTensor>>& allTensors);
42 OH_NN_ReturnCode SetBeginMask(const std::shared_ptr<NNTensor>& tensor);
43 OH_NN_ReturnCode SetEndMask(const std::shared_ptr<NNTensor>& tensor);
44 OH_NN_ReturnCode SetEllipsisMask(const std::shared_ptr<NNTensor>& tensor);
45 OH_NN_ReturnCode SetNewAxisMask(const std::shared_ptr<NNTensor>& tensor);
46 OH_NN_ReturnCode SetShrinkAxisMask(const std::shared_ptr<NNTensor>& tensor);
H A Dpooling_builder.h28 typedef OH_NN_ReturnCode (PoolingBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);
36 const std::vector<std::shared_ptr<NNTensor>>& allTensors);
40 const std::vector<std::shared_ptr<NNTensor>>& allTensors);
42 OH_NN_ReturnCode SetKernel(const std::shared_ptr<NNTensor>& tensor);
43 OH_NN_ReturnCode SetStrides(const std::shared_ptr<NNTensor>& tensor);
44 OH_NN_ReturnCode SetPadModeOrPaddings(const std::shared_ptr<NNTensor>& tensor);
45 OH_NN_ReturnCode SetRoundMode(const std::shared_ptr<NNTensor>& tensor);
46 OH_NN_ReturnCode SetActivation(const std::shared_ptr<NNTensor>& tensor);
47 OH_NN_ReturnCode SetGlobal(const std::shared_ptr<NNTensor>& tensor);
H A Dlstm_builder.cpp31 OH_NN_ReturnCode LSTMBuilder::SetBidirectional(const std::shared_ptr<NNTensor>& tensor) in SetBidirectional()
53 OH_NN_ReturnCode LSTMBuilder::SetHasBias(const std::shared_ptr<NNTensor>& tensor) in SetHasBias()
75 OH_NN_ReturnCode LSTMBuilder::SetInputSize(const std::shared_ptr<NNTensor>& tensor) in SetInputSize()
97 OH_NN_ReturnCode LSTMBuilder::SetHiddenSize(const std::shared_ptr<NNTensor>& tensor) in SetHiddenSize()
119 OH_NN_ReturnCode LSTMBuilder::SetNumLayers(const std::shared_ptr<NNTensor>& tensor) in SetNumLayers()
141 OH_NN_ReturnCode LSTMBuilder::SetNumDirections(const std::shared_ptr<NNTensor>& tensor) in SetNumDirections()
163 OH_NN_ReturnCode LSTMBuilder::SetDropout(const std::shared_ptr<NNTensor>& tensor) in SetDropout()
185 OH_NN_ReturnCode LSTMBuilder::SetZoneoutCell(const std::shared_ptr<NNTensor>& tensor) in SetZoneoutCell()
207 OH_NN_ReturnCode LSTMBuilder::SetZoneoutHidden(const std::shared_ptr<NNTensor>& tensor) in SetZoneoutHidden()
229 OH_NN_ReturnCode LSTMBuilder::SetProjSize(const std::shared_ptr<NNTensor>& tensor) in SetProjSize()
[all …]
H A Dlrn_builder.h26 typedef OH_NN_ReturnCode (LRNBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);
33 const std::vector<std::shared_ptr<NNTensor>>& allTensors) override;
38 OH_NN_ReturnCode SetDepthRadius(const std::shared_ptr<NNTensor>& tensor);
39 OH_NN_ReturnCode SetBias(const std::shared_ptr<NNTensor>& tensor);
40 OH_NN_ReturnCode SetAlpha(const std::shared_ptr<NNTensor>& tensor);
41 OH_NN_ReturnCode SetBeta(const std::shared_ptr<NNTensor>& tensor);
42 OH_NN_ReturnCode SetNormRegion(const std::shared_ptr<NNTensor>& tensor);
H A Dfullconnection_builder.h29 typedef OH_NN_ReturnCode (FullConnectionBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);
36 const std::vector<std::shared_ptr<NNTensor>>& allTensors) override;
43 … const std::vector<std::shared_ptr<NNTensor>>& allTensors);
44 OH_NN_ReturnCode SetHasBias(const std::shared_ptr<NNTensor>& tensor);
45 OH_NN_ReturnCode SetUseAxis(const std::shared_ptr<NNTensor>& tensor);
46 OH_NN_ReturnCode SetFullConnectionActivation(const std::shared_ptr<NNTensor>& tensor);
47 OH_NN_ReturnCode SetAxis(const std::shared_ptr<NNTensor>& tensor);
H A Dresize_bilinear_builder.h28 typedef OH_NN_ReturnCode (ResizeBilinearBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);
35 const std::vector<std::shared_ptr<NNTensor>>& allTensors) override;
40 OH_NN_ReturnCode SetNewHeight(const std::shared_ptr<NNTensor>& tensor);
41 OH_NN_ReturnCode SetNewWidth(const std::shared_ptr<NNTensor>& tensor);
42 OH_NN_ReturnCode SetPreserveAspectRatio(const std::shared_ptr<NNTensor>& tensor);
43 OH_NN_ReturnCode SetCoordinateTransformMode(const std::shared_ptr<NNTensor>& tensor);
44 OH_NN_ReturnCode SetExcludeOutside(const std::shared_ptr<NNTensor>& tensor);
H A Ddetection_post_process_builder.cpp36 OH_NN_ReturnCode DetectionPostProcessBuilder::SetInputSize(const std::shared_ptr<NNTensor>& tensor) in SetInputSize()
58 OH_NN_ReturnCode DetectionPostProcessBuilder::SetScale(const std::shared_ptr<NNTensor>& tensor) in SetScale()
84 OH_NN_ReturnCode DetectionPostProcessBuilder::SetNmsIoUThreshold(const std::shared_ptr<NNTensor>& t… in SetNmsIoUThreshold()
106 OH_NN_ReturnCode DetectionPostProcessBuilder::SetNmsScoreThreshold(const std::shared_ptr<NNTensor>&… in SetNmsScoreThreshold()
128 OH_NN_ReturnCode DetectionPostProcessBuilder::SetMaxDetections(const std::shared_ptr<NNTensor>& ten… in SetMaxDetections()
150 OH_NN_ReturnCode DetectionPostProcessBuilder::SetDetectionsPerClass(const std::shared_ptr<NNTensor>… in SetDetectionsPerClass()
194 OH_NN_ReturnCode DetectionPostProcessBuilder::SetNumClasses(const std::shared_ptr<NNTensor>& tensor) in SetNumClasses()
216 OH_NN_ReturnCode DetectionPostProcessBuilder::SetUseRegularNms(const std::shared_ptr<NNTensor>& ten… in SetUseRegularNms()
238 OH_NN_ReturnCode DetectionPostProcessBuilder::SetOutQuantized(const std::shared_ptr<NNTensor>& tens… in SetOutQuantized()
264 … const std::vector<std::shared_ptr<NNTensor>>& allTensors) in Build()
[all …]
H A Dargmax_builder.h29 typedef OH_NN_ReturnCode (ArgMaxBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);
36 const std::vector<std::shared_ptr<NNTensor>>& allTensors) override;
40 OH_NN_ReturnCode SetAxis(const std::shared_ptr<NNTensor>& tensor);
41 OH_NN_ReturnCode SetKeepdims(const std::shared_ptr<NNTensor>& tensor);
42 OH_NN_ReturnCode SetTopK(const std::shared_ptr<NNTensor>& tensor);
43 OH_NN_ReturnCode SetOutMaxValue(const std::shared_ptr<NNTensor>& tensor);
H A Dlayernorm_builder.h26 typedef OH_NN_ReturnCode (LayerNormBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);
33 const std::vector<std::shared_ptr<NNTensor>>& allTensors) override;
37 OH_NN_ReturnCode SetBeginNormAxis(const std::shared_ptr<NNTensor>& tensor);
38 OH_NN_ReturnCode SetEpsilon(const std::shared_ptr<NNTensor>& tensor);
39 OH_NN_ReturnCode SetBeginParamsAxis(const std::shared_ptr<NNTensor>& tensor);
41 int64_t beginAxis, const std::vector<std::shared_ptr<NNTensor>>& allTensors) const;
H A Dsplit_builder.h29 typedef OH_NN_ReturnCode (SplitBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);
36 const std::vector<std::shared_ptr<NNTensor>>& allTensors) override;
43 const std::vector<std::shared_ptr<NNTensor>>& allTensors);
44 OH_NN_ReturnCode SetAxis(const std::shared_ptr<NNTensor>& tensor);
45 OH_NN_ReturnCode SetOutputNum(const std::shared_ptr<NNTensor>& tensor);
46 OH_NN_ReturnCode SetSizeSplits(const std::shared_ptr<NNTensor>& tensor);
H A Dstrided_slice_builder.cpp36 … const std::vector<std::shared_ptr<NNTensor>>& allTensors) in SetInputOutput()
50 OH_NN_ReturnCode StridedSliceBuilder::SetBeginMask(const std::shared_ptr<NNTensor>& tensor) in SetBeginMask()
67 OH_NN_ReturnCode StridedSliceBuilder::SetEndMask(const std::shared_ptr<NNTensor>& tensor) in SetEndMask()
84 OH_NN_ReturnCode StridedSliceBuilder::SetEllipsisMask(const std::shared_ptr<NNTensor>& tensor) in SetEllipsisMask()
101 OH_NN_ReturnCode StridedSliceBuilder::SetNewAxisMask(const std::shared_ptr<NNTensor>& tensor) in SetNewAxisMask()
118 OH_NN_ReturnCode StridedSliceBuilder::SetShrinkAxisMask(const std::shared_ptr<NNTensor>& tensor) in SetShrinkAxisMask()
144 … const std::vector<std::shared_ptr<NNTensor>>& allTensors) in Build()
164 std::shared_ptr<NNTensor> tensor = allTensors[i]; in Build()
H A Dexp_builder.h29 typedef OH_NN_ReturnCode (ExpBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);
36 const std::vector<std::shared_ptr<NNTensor>>& allTensors) override;
41 OH_NN_ReturnCode SetBase(const std::shared_ptr<NNTensor>& tensor);
42 OH_NN_ReturnCode SetScale(const std::shared_ptr<NNTensor>& tensor);
43 OH_NN_ReturnCode SetShift(const std::shared_ptr<NNTensor>& tensor);
H A DreduceL2_builder.h26 typedef OH_NN_ReturnCode (ReduceL2Builder::*FuncPtr)(const std::shared_ptr<NNTensor>&);
33 const std::vector<std::shared_ptr<NNTensor>>& allTensors) override;
38 OH_NN_ReturnCode SetCoeff(const std::shared_ptr<NNTensor>& tensor);
39 OH_NN_ReturnCode SetReduceToEnd(const std::shared_ptr<NNTensor>& tensor);
40 OH_NN_ReturnCode SetKeepDims(const std::shared_ptr<NNTensor>& tensor);
H A Dreduceall_builder.h26 typedef OH_NN_ReturnCode (ReduceAllBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);
33 const std::vector<std::shared_ptr<NNTensor>>& allTensors) override;
38 OH_NN_ReturnCode SetCoeff(const std::shared_ptr<NNTensor>& tensor);
39 OH_NN_ReturnCode SetReduceToEnd(const std::shared_ptr<NNTensor>& tensor);
40 OH_NN_ReturnCode SetKeepDims(const std::shared_ptr<NNTensor>& tensor);
H A Dreducemax_builder.h26 typedef OH_NN_ReturnCode (ReduceMaxBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);
33 const std::vector<std::shared_ptr<NNTensor>>& allTensors) override;
38 OH_NN_ReturnCode SetCoeff(const std::shared_ptr<NNTensor>& tensor);
39 OH_NN_ReturnCode SetReduceToEnd(const std::shared_ptr<NNTensor>& tensor);
40 OH_NN_ReturnCode SetKeepDims(const std::shared_ptr<NNTensor>& tensor);
H A Dreducemean_builder.h26 typedef OH_NN_ReturnCode (ReduceMeanBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);
33 const std::vector<std::shared_ptr<NNTensor>>& allTensors) override;
38 OH_NN_ReturnCode SetCoeff(const std::shared_ptr<NNTensor>& tensor);
39 OH_NN_ReturnCode SetReduceToEnd(const std::shared_ptr<NNTensor>& tensor);
40 OH_NN_ReturnCode SetKeepDims(const std::shared_ptr<NNTensor>& tensor);

12345678910>>...14