Home
last modified time | relevance | path

Searched defs:tensor (Results 1 – 25 of 73) sorted by relevance

123

/foundation/ai/neural_network_runtime/test/unittest/ops/
Dconv2d_tranpose_padmode_test.cpp58 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetPadMode() local
69 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetOutPaddings() local
229 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_stride_dim, nullptr, variable
259 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_dilation_dim, nullptr, variable
288 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
347 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
376 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
404 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, outPaddingsDim, nullptr, variable
432 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, outPaddingsDim, nullptr, variable
489 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
[all …]
Dconv2d_pad_test.cpp57 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetPad() local
218 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_stride_dim, nullptr, OH_NN_CONV2… variable
245 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_dilation_dim, nullptr, OH_NN_CON… variable
271 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_pad_dim, nullptr, OH_NN_CONV2D_P… variable
298 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, OH_NN_CONV2D… variable
325 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, OH_NN_CONV2D… variable
351 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, groupDim, nullptr, OH_NN_CONV2D_GR… variable
379 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, activationDim, nullptr, variable
400 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_stride_dim, nullptr, variable
425 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_dilation_dim, nullptr, variable
[all …]
Dmaxpool_pad_test.cpp57 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetPad() local
187 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_kenelsize_dim, nullptr, variable
215 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_stride_dim, nullptr, OH_NN_MAX_P… variable
240 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_pad_dim, nullptr, OH_NN_MAX_POOL… variable
265 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
293 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
311 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_kenelsize_dim, nullptr, variable
333 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_stride_dim, nullptr, OH_NN_MAX_P… variable
354 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_MAX_POOL… variable
375 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
Davgpool_pad_test.cpp57 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetPad() local
189 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_kenelsize_dim, nullptr, variable
215 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_stride_dim, nullptr, OH_NN_AVG_P… variable
240 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_pad_dim, nullptr, OH_NN_AVG_POOL… variable
265 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
293 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
311 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_kenelsize_dim, nullptr, variable
333 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_stride_dim, nullptr, OH_NN_AVG_P… variable
354 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_AVG_POOL… variable
376 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
Davgpool_padmod_test.cpp55 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetPadMode() local
189 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_kenelsize_dim, nullptr, variable
215 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_stride_dim, nullptr, OH_NN_AVG_P… variable
243 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, OH_NN_AVG_PO… variable
266 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
294 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
318 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
341 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, OH_NN_AVG_POO… variable
365 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
Dconv2d_transpose_pad_test.cpp60 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetPad() local
72 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetOutPaddings() local
96 std::shared_ptr<NNTensor> tensor; in SetConv2dTransposeInput() local
236 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_stride_dim, nullptr, variable
263 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_dilation_dim, nullptr, variable
290 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_pad_dim, nullptr, OH_NN_CONV2D_T… variable
317 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_outpaddings_dim, nullptr, variable
344 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, OH_NN_CONV2D… variable
370 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
395 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_outpaddings_dim, nullptr, variable
[all …]
Dconv2d_padmode_test.cpp55 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetPadMode() local
216 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_stride_dim, nullptr, OH_NN_CONV2… variable
244 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_dilation_dim, nullptr, OH_NN_CON… variable
276 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, OH_NN_CONV2D… variable
303 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, OH_NN_CONV2D… variable
331 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, OH_NN_CONV2D… variable
357 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, groupDim, nullptr, OH_NN_CONV2D_GR… variable
385 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, activationDim, nullptr, variable
413 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, activationDim, nullptr, variable
437 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, OH_NN_CONV2D_… variable
[all …]
Dfullconnection_with_axis_test.cpp55 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_FLOAT32, m_input_dim, nullptr, OH_NN_TENS… in SetInputToAlltensor() local
75 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetActivation() local
86 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SeAxis() local
177 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, OH_NN_FULL_C… variable
200 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
223 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, paramDimTest, nullptr, variable
248 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, paramDimTest, nullptr, variable
270 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_param_dim, nullptr, OH_NN_FULL_C… variable
289 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
Dargmax_test.cpp54 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetArgmaxAxis() local
64 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetArgmaxKeepdims() local
199 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_FLOAT32, m_param_dim, nullptr, OH_NN_ARG_… variable
221 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_param_dim, nullptr, OH_NN_ARG_MA… variable
242 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_param_dim, nullptr, OH_NN_AVG_PO… variable
262 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_param_dim, nullptr, OH_NN_ARG_MA… variable
280 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_ARG_MAX… variable
Ddepthwise_conv2d_native_pad_test.cpp57 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetPad() local
219 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_stride_dim, nullptr, variable
245 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_dilation_dim, nullptr, variable
274 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_pad_dim, nullptr, variable
300 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
327 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, activationDim, nullptr, variable
353 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, activationDim, nullptr, variable
379 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, activationDim, nullptr, variable
406 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_pad_dim, nullptr, variable
500 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_stride_dim, nullptr, variable
[all …]
Deltwise_test.cpp51 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetEltwiseMode() local
171 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, OH_NN_ELTWIS… variable
194 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, OH_NN_ELTWISE… variable
215 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, OH_NN_ELTWISE… variable
236 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, OH_NN_DIV_ACT… variable
254 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, OH_NN_ELTWISE… variable
Ddiv_test.cpp53 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveParamsTensor() local
166 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, OH_NN_DIV_AC… variable
186 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, OH_NN_DIV_ACT… variable
206 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, OH_NN_DIV_ACT… variable
225 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, OH_NN_ADD_ACT… variable
244 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
Dbatch_to_space_nd_test.cpp55 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetBlockSize() local
67 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetCrops() local
199 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_crops_dim, nullptr, variable
221 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_block_dim, nullptr, variable
244 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_block_dim, nullptr, variable
266 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_block_dim, nullptr, variable
286 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_crops_dim, nullptr, variable
Dfullconnection_test.cpp54 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_FLOAT32, m_input_dim, nullptr, OH_NN_TENS… in SetInputToAlltensor() local
76 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetActivation() local
168 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
192 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
215 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
238 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
Dadd_test.cpp53 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveParamsTensor() local
172 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, OH_NN_ADD_AC… variable
191 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, OH_NN_ADD_ACT… variable
211 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, OH_NN_ARG_MAX… variable
231 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, OH_NN_ADD_ACT… variable
Dmaxpool_padmode_test.cpp55 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetPadMode() local
183 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_kenelsize_dim, nullptr, variable
209 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_stride_dim, nullptr, OH_NN_MAX_P… variable
236 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, OH_NN_MAX_PO… variable
258 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
286 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
Ddepthwise_conv2d_native_padmode_test.cpp55 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetPadMode() local
217 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_stride_dim, nullptr, variable
244 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_dilation_dim, nullptr, variable
271 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
297 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
323 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, activationDim, nullptr, variable
348 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
Dconcat_two_inputs_test.cpp52 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetAxis() local
178 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, OH_NN_CONCAT… variable
200 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT… variable
220 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT… variable
Dconcat_three_inputs_test.cpp52 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetAxis() local
153 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, OH_NN_CONCAT… variable
175 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT… variable
196 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONV2D… variable
/foundation/ai/neural_network_runtime/test/unittest/components/inner_model/
Dnn_tensor_test.cpp40 OH_NN_Tensor tensor{OH_NN_FLOAT32, 2, dim, nullptr, OH_NN_TENSOR}; variable
53 OH_NN_Tensor tensor; variable
72 OH_NN_Tensor tensor{OH_NN_FLOAT32, 2, dim, nullptr, OH_NN_TENSOR}; variable
86 OH_NN_Tensor tensor{OH_NN_FLOAT32, 2, dim, nullptr, OH_NN_TENSOR}; variable
100 OH_NN_Tensor tensor{OH_NN_FLOAT32, 2, dim, nullptr, OH_NN_TENSOR}; variable
121 OH_NN_Tensor tensor{OH_NN_FLOAT32, 2, dim, &quantParam, OH_NN_TENSOR}; variable
140 OH_NN_Tensor tensor{OH_NN_FLOAT32, 2, dim, &quantParam, OH_NN_TENSOR}; variable
158 OH_NN_Tensor tensor{OH_NN_FLOAT32, 2, dim, &quantParam, OH_NN_TENSOR}; variable
176 OH_NN_Tensor tensor{OH_NN_FLOAT32, 2, dim, &quantParam, OH_NN_TENSOR}; variable
189 OH_NN_Tensor tensor{OH_NN_FLOAT32, 2, dim, nullptr, OH_NN_TENSOR}; variable
[all …]
/foundation/ai/neural_network_runtime/frameworks/native/ops/
Dresize_bilinear_builder.cpp32 OH_NN_ReturnCode ResizeBilinearBuilder::SetNewHeight(std::shared_ptr<NNTensor> tensor) in SetNewHeight()
55 OH_NN_ReturnCode ResizeBilinearBuilder::SetNewWidth(std::shared_ptr<NNTensor> tensor) in SetNewWidth()
78 OH_NN_ReturnCode ResizeBilinearBuilder::SetPreserveAspectRatio(std::shared_ptr<NNTensor> tensor) in SetPreserveAspectRatio()
101 OH_NN_ReturnCode ResizeBilinearBuilder::SetCoordinateTransformMode(std::shared_ptr<NNTensor> tensor) in SetCoordinateTransformMode()
127 OH_NN_ReturnCode ResizeBilinearBuilder::SetExcludeOutside(std::shared_ptr<NNTensor> tensor) in SetExcludeOutside()
170 std::shared_ptr<NNTensor> tensor = allTensors[i]; in Build() local
Dstrided_slice_builder.cpp49 OH_NN_ReturnCode StridedSliceBuilder::SetBeginMask(std::shared_ptr<NNTensor> tensor) in SetBeginMask()
66 OH_NN_ReturnCode StridedSliceBuilder::SetEndMask(std::shared_ptr<NNTensor> tensor) in SetEndMask()
83 OH_NN_ReturnCode StridedSliceBuilder::SetEllipsisMask(std::shared_ptr<NNTensor> tensor) in SetEllipsisMask()
100 OH_NN_ReturnCode StridedSliceBuilder::SetNewAxisMask(std::shared_ptr<NNTensor> tensor) in SetNewAxisMask()
117 OH_NN_ReturnCode StridedSliceBuilder::SetShrinkAxisMask(std::shared_ptr<NNTensor> tensor) in SetShrinkAxisMask()
157 std::shared_ptr<NNTensor> tensor = allTensors[i]; in Build() local
Dconv2d_transpose_builder.cpp77 OH_NN_ReturnCode Conv2DTransposeBuilder::SetStrides(std::shared_ptr<NNTensor> tensor) in SetStrides()
98 OH_NN_ReturnCode Conv2DTransposeBuilder::SetDilation(std::shared_ptr<NNTensor> tensor) in SetDilation()
119 OH_NN_ReturnCode Conv2DTransposeBuilder::SetPad(std::shared_ptr<NNTensor> tensor) in SetPad()
164 OH_NN_ReturnCode Conv2DTransposeBuilder::SetGroup(std::shared_ptr<NNTensor> tensor) in SetGroup()
188 OH_NN_ReturnCode Conv2DTransposeBuilder::SetOutPadding(std::shared_ptr<NNTensor> tensor) in SetOutPadding()
209 OH_NN_ReturnCode Conv2DTransposeBuilder::SetActivation(std::shared_ptr<NNTensor> tensor) in SetActivation()
256 std::shared_ptr<NNTensor> tensor = allTensors[i]; // 参数 tensor in Build() local
Dpooling_builder.cpp48 std::shared_ptr<NNTensor> tensor = allTensors[i]; in PoolingBuild() local
100 OH_NN_ReturnCode PoolingBuilder::SetKernel(std::shared_ptr<NNTensor> tensor) in SetKernel()
122 OH_NN_ReturnCode PoolingBuilder::SetStrides(std::shared_ptr<NNTensor> tensor) in SetStrides()
144 OH_NN_ReturnCode PoolingBuilder::SetPadModeOrPaddings(std::shared_ptr<NNTensor> tensor) in SetPadModeOrPaddings()
188 OH_NN_ReturnCode PoolingBuilder::SetActivation(std::shared_ptr<NNTensor> tensor) in SetActivation()
Dmatmul_builder.cpp34 OH_NN_ReturnCode MatmulBuilder::SetTransposeA(std::shared_ptr<NNTensor> tensor) in SetTransposeA()
57 OH_NN_ReturnCode MatmulBuilder::SetTransposeB(std::shared_ptr<NNTensor> tensor) in SetTransposeB()
80 OH_NN_ReturnCode MatmulBuilder::SetActivationType(std::shared_ptr<NNTensor> tensor) in SetActivationType()
130 std::shared_ptr<NNTensor> tensor = allTensors[i]; in Build() local

123