| /ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/ |
| H A D | lstm_builder.cpp | 24 static const int SCALAR_LENGTH = 1; variable
38 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetBidirectional()
60 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetHasBias()
82 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetInputSize()
104 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetHiddenSize()
126 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetNumLayers()
148 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetNumDirections()
170 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetDropout()
192 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetZoneoutCell()
214 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetZoneoutHidden()
[all …]
|
| H A D | detection_post_process_builder.cpp | 29 static const int SCALAR_LENGTH = 1; variable
43 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetInputSize()
91 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetNmsIoUThreshold()
113 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetNmsScoreThreshold()
135 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetMaxDetections()
157 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetDetectionsPerClass()
179 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetMaxClassesPerDetection()
201 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetNumClasses()
223 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetUseRegularNms()
245 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetOutQuantized()
|
| H A D | lrn_builder.cpp | 27 static const int SCALAR_LENGTH = 1; variable
42 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetDepthRadius()
64 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetBias()
86 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetAlpha()
108 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetBeta()
130 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetNormRegion()
|
| H A D | fullconnection_builder.cpp | 26 static constexpr int SCALAR_LENGTH = 1; variable
64 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetHasBias()
86 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetUseAxis()
112 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetFullConnectionActivation()
142 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetAxis()
|
| H A D | exp_builder.cpp | 24 static const int SCALAR_LENGTH = 1; variable
38 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetBase()
60 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetScale()
82 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetShift()
|
| H A D | range_builder.cpp | 24 static const int SCALAR_LENGTH = 1; variable
38 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetStart()
60 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetLimit()
82 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetDelta()
|
| H A D | clip_builder.cpp | 24 static constexpr int SCALAR_LENGTH = 1; variable
38 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetMax()
60 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetMin()
|
| H A D | pow_builder.cpp | 27 static const int SCALAR_LENGTH = 1; variable
41 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetScale()
63 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetShift()
|
| H A D | l2_normalize_builder.cpp | 28 static const int SCALAR_LENGTH = 1; variable
68 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetEpsilon()
90 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetActivationType()
|
| H A D | flatten_builder.cpp | 24 static const int SCALAR_LENGTH = 1; variable
38 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetAxis()
|
| H A D | instance_norm_builder.cpp | 24 static const int SCALAR_LENGTH = 1; variable
38 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetEpsilon()
|
| H A D | log_softmax_builder.cpp | 27 static const int SCALAR_LENGTH = 1; variable
41 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetAxis()
|
| H A D | all_builder.cpp | 24 static const int SCALAR_LENGTH = 1; variable
38 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetKeepDims()
|
| H A D | assert_builder.cpp | 24 static const int SCALAR_LENGTH = 1; variable
38 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetSummarize()
|
| H A D | batchnorm_builder.cpp | 28 static const int SCALAR_LENGTH = 1; variable
43 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetEpsilon()
|
| H A D | space_to_depth_builder.cpp | 29 static const int SCALAR_LENGTH = 1; variable
43 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetBlockSize()
|
| H A D | div_builder.cpp | 27 static constexpr int SCALAR_LENGTH = 1; variable
38 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetActicationType()
|
| H A D | eltwise_builder.cpp | 24 static constexpr int SCALAR_LENGTH = 1; variable
40 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetMode()
|
| H A D | gelu_builder.cpp | 27 static const int SCALAR_LENGTH = 1; variable
41 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetApproximate()
|
| H A D | leaky_relu_builder.cpp | 24 static const int SCALAR_LENGTH = 1; variable
38 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetNegativeSlope()
|
| H A D | constant_of_shape_builder.cpp | 24 static const int SCALAR_LENGTH = 1; variable
38 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetDataType()
|
| H A D | crop_builder.cpp | 27 static const int SCALAR_LENGTH = 1; variable
41 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetAxis()
|
| H A D | top_k_builder.cpp | 27 static const int SCALAR_LENGTH = 1; variable
57 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetAxis()
|
| H A D | unstack_builder.cpp | 24 static const int SCALAR_LENGTH = 1; variable
38 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetAxis()
|
| H A D | depth_to_space_builder.cpp | 27 static const int SCALAR_LENGTH = 1; variable
42 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetBlockSize()
|