| /ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/ |
| H A D | detection_post_process_builder.cpp | 43 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetInputSize()
75 uint32_t elementCount = tensor->GetElementCount(); in SetScale()
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 | lstm_builder.cpp | 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()
236 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetProjSize()
|
| H A D | conv2d_transpose_builder.cpp | 94 uint32_t elementSize = tensor->GetElementCount(); in SetStrides()
115 uint32_t dilationSize = tensor->GetElementCount(); in SetDilation()
126 if (tensor->GetElementCount() == PAD_MODE_PARAM_NUM) { in SetPad()
128 } else if (tensor->GetElementCount() != PAD_LIST_PARAM_NUM) { in SetPad()
159 uint32_t padListPadSize = tensor->GetElementCount(); in SetPad()
170 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetGroup()
205 uint32_t outputPadSize = tensor->GetElementCount(); in SetOutPadding()
215 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetActivation()
|
| H A D | conv2d_builder.cpp | 101 uint32_t stridesSize = tensor->GetElementCount(); in SetStrides()
122 uint32_t dilationSize = tensor->GetElementCount(); in SetDilation()
133 if (tensor->GetElementCount() == PAD_MODE_GET) { in SetPad()
135 } else if (tensor->GetElementCount() != PAD_LIST_GET) { in SetPad()
166 uint32_t padListSize = tensor->GetElementCount(); in SetPad()
177 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetGroup()
201 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetActavitation()
|
| H A D | lrn_builder.cpp | 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 | pooling_builder.cpp | 111 uint32_t kernelSize = tensor->GetElementCount(); in SetKernel()
133 uint32_t strideslSize = tensor->GetElementCount(); in SetStrides()
148 size_t tensorElementCount = tensor->GetElementCount(); in SetPadModeOrPaddings()
187 if (tensor->GetElementCount() != ACTIVATION_LENGTH) { in SetRoundMode()
220 if (tensor->GetElementCount() != ACTIVATION_LENGTH) { in SetActivation()
254 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetGlobal()
|
| H A D | resize_bilinear_builder.cpp | 36 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetNewHeight()
59 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetNewWidth()
82 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetPreserveAspectRatio()
105 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetCoordinateTransformMode()
131 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetExcludeOutside()
|
| H A D | depthwise_conv2d_native_builder.cpp | 47 if (tensor->GetElementCount() == PAD_MODE_SIZE) { in SetIsPadMode()
49 } else if (tensor->GetElementCount() != PAD_LIST_SIZE) { in SetIsPadMode()
62 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetActivation()
118 uint32_t stridesSize = tensor->GetElementCount(); in SetStrides()
137 uint32_t dilationSize = tensor->GetElementCount(); in SetDilation()
178 uint32_t padListSize = tensor->GetElementCount(); in SetPadModeOrPaddings()
|
| H A D | fullconnection_builder.cpp | 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 | 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 | 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 | reduceL2_builder.cpp | 41 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetCoeff()
64 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetReduceToEnd()
87 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetKeepDims()
|
| H A D | reduceall_builder.cpp | 41 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetCoeff()
64 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetReduceToEnd()
87 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetKeepDims()
|
| H A D | reducemax_builder.cpp | 41 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetCoeff()
64 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetReduceToEnd()
87 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetKeepDims()
|
| H A D | reducemean_builder.cpp | 41 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetCoeff()
64 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetReduceToEnd()
82 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetKeepDims()
|
| H A D | reducemin_builder.cpp | 41 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetCoeff()
64 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetReduceToEnd()
87 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetKeepDims()
|
| H A D | reduceprod_builder.cpp | 41 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetCoeff()
64 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetReduceToEnd()
82 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetKeepDims()
|
| H A D | reducesum_builder.cpp | 41 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetCoeff()
64 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetReduceToEnd()
87 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetKeepDims()
|
| H A D | l2_normalize_builder.cpp | 52 uint32_t elementCount = tensor->GetElementCount(); in SetAxis()
68 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetEpsilon()
90 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetActivationType()
|
| H A D | matmul_builder.cpp | 38 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetTransposeA()
61 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetTransposeB()
84 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetActivationType()
|
| H A D | space_to_batch_nd_builder.cpp | 51 if (tensor->GetElementCount() != BLOCK_SIZE) { in SetBlockShape()
63 const uint32_t elementSize = tensor->GetElementCount(); in SetBlockShape()
84 if (tensor->GetElementCount() != PADDINGS_SIZE) { in SetPaddings()
|
| H A D | split_builder.cpp | 71 if (tensor->GetElementCount() != 1) { in SetAxis()
93 if (tensor->GetElementCount() != 1) { in SetOutputNum()
111 for (uint32_t i = 0; i < tensor->GetElementCount(); i++) { in SetSizeSplits()
|
| H A D | clip_builder.cpp | 38 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetMax()
60 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetMin()
|
| H A D | constant_of_shape_builder.cpp | 38 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetDataType()
70 uint32_t elementCount = tensor->GetElementCount(); in SetValue()
|
| H A D | crop_builder.cpp | 41 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetAxis()
73 uint32_t elementCount = tensor->GetElementCount(); in SetOffset()
|