Home
last modified time | relevance | path

Searched refs:tensor (Results 1 – 25 of 334) sorted by relevance

12345678910>>...14

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Ddetection_post_process_builder.cpp48 void* buffer = tensor->GetBuffer(); in SetInputSize()
67 void* buffer = tensor->GetBuffer(); in SetScale()
96 void* buffer = tensor->GetBuffer(); in SetNmsIoUThreshold()
118 void* buffer = tensor->GetBuffer(); in SetNmsScoreThreshold()
140 void* buffer = tensor->GetBuffer(); in SetMaxDetections()
162 void* buffer = tensor->GetBuffer(); in SetDetectionsPerClass()
184 void* buffer = tensor->GetBuffer(); in SetMaxClassesPerDetection()
206 void* buffer = tensor->GetBuffer(); in SetNumClasses()
228 void* buffer = tensor->GetBuffer(); in SetUseRegularNms()
289 tensor->IdentifyOpParameter(); in Build()
[all …]
H A Dlstm_builder.cpp43 void* buffer = tensor->GetBuffer(); in SetBidirectional()
65 void* buffer = tensor->GetBuffer(); in SetHasBias()
87 void* buffer = tensor->GetBuffer(); in SetInputSize()
109 void* buffer = tensor->GetBuffer(); in SetHiddenSize()
131 void* buffer = tensor->GetBuffer(); in SetNumLayers()
153 void* buffer = tensor->GetBuffer(); in SetNumDirections()
175 void* buffer = tensor->GetBuffer(); in SetDropout()
197 void* buffer = tensor->GetBuffer(); in SetZoneoutCell()
219 void* buffer = tensor->GetBuffer(); in SetZoneoutHidden()
257 tensor->IdentifyOpParameter(); in ParseParam()
[all …]
H A Dresize_bilinear_builder.cpp35 tensor->IdentifyOpParameter(); in SetNewHeight()
46 void* buffer = tensor->GetBuffer(); in SetNewHeight()
58 tensor->IdentifyOpParameter(); in SetNewWidth()
69 void* buffer = tensor->GetBuffer(); in SetNewWidth()
81 tensor->IdentifyOpParameter(); in SetPreserveAspectRatio()
92 void* buffer = tensor->GetBuffer(); in SetPreserveAspectRatio()
104 tensor->IdentifyOpParameter(); in SetCoordinateTransformMode()
117 void* buffer = tensor->GetBuffer(); in SetCoordinateTransformMode()
130 tensor->IdentifyOpParameter(); in SetExcludeOutside()
141 void* buffer = tensor->GetBuffer(); in SetExcludeOutside()
[all …]
H A Dpooling_builder.cpp61 returnCode = (this->*(m_paramMap[tensor->GetType()]))(tensor); in PoolingBuild()
97 tensor->IdentifyOpParameter(); in SetKernel()
104 void* buffer = tensor->GetBuffer(); in SetKernel()
119 tensor->IdentifyOpParameter(); in SetStrides()
126 void* buffer = tensor->GetBuffer(); in SetStrides()
141 tensor->IdentifyOpParameter(); in SetPadModeOrPaddings()
143 void* buffer = tensor->GetBuffer(); in SetPadModeOrPaddings()
185 tensor->IdentifyOpParameter(); in SetRoundMode()
197 void* buffer = tensor->GetBuffer(); in SetRoundMode()
218 tensor->IdentifyOpParameter(); in SetActivation()
[all …]
H A Dconv2d_transpose_builder.cpp81 tensor->IdentifyOpParameter(); in SetStrides()
88 void* buffer = tensor->GetBuffer(); in SetStrides()
102 tensor->IdentifyOpParameter(); in SetDilation()
109 void* buffer = tensor->GetBuffer(); in SetDilation()
123 tensor->IdentifyOpParameter(); in SetPad()
133 void* buffer = tensor->GetBuffer(); in SetPad()
168 tensor->IdentifyOpParameter(); in SetGroup()
180 void* buffer = tensor->GetBuffer(); in SetGroup()
192 tensor->IdentifyOpParameter(); in SetOutPadding()
213 tensor->IdentifyOpParameter(); in SetActivation()
[all …]
H A Dlrn_builder.cpp37 if (tensor->GetDataType() != OH_NN_INT64) { in SetDepthRadius()
47 void* buffer = tensor->GetBuffer(); in SetDepthRadius()
59 if (tensor->GetDataType() != OH_NN_FLOAT32) { in SetBias()
69 void* buffer = tensor->GetBuffer(); in SetBias()
81 if (tensor->GetDataType() != OH_NN_FLOAT32) { in SetAlpha()
91 void* buffer = tensor->GetBuffer(); in SetAlpha()
113 void* buffer = tensor->GetBuffer(); in SetBeta()
125 if (tensor->GetDataType() != OH_NN_INT32) { in SetNormRegion()
135 void* buffer = tensor->GetBuffer(); in SetNormRegion()
180 tensor->IdentifyOpParameter(); in Build()
[all …]
H A Dconv2d_builder.cpp88 tensor->IdentifyOpParameter(); in SetStrides()
95 void* buffer = tensor->GetBuffer(); in SetStrides()
109 tensor->IdentifyOpParameter(); in SetDilation()
116 void* buffer = tensor->GetBuffer(); in SetDilation()
130 tensor->IdentifyOpParameter(); in SetPad()
140 void* buffer = tensor->GetBuffer(); in SetPad()
175 tensor->IdentifyOpParameter(); in SetGroup()
187 void* buffer = tensor->GetBuffer(); in SetGroup()
199 tensor->IdentifyOpParameter(); in SetActavitation()
211 void* buffer = tensor->GetBuffer(); in SetActavitation()
[all …]
H A Ddepthwise_conv2d_native_builder.cpp60 tensor->IdentifyOpParameter(); in SetActivation()
67 if (tensor->GetDataType() != OH_NN_INT8) { in SetActivation()
72 void* buffer = tensor->GetBuffer(); in SetActivation()
106 tensor->IdentifyOpParameter(); in SetStrides()
107 if (tensor->GetDataType() != OH_NN_INT64) { in SetStrides()
112 void* buffer = tensor->GetBuffer(); in SetStrides()
125 tensor->IdentifyOpParameter(); in SetDilation()
131 void* buffer = tensor->GetBuffer(); in SetDilation()
145 tensor->IdentifyOpParameter(); in SetPadModeOrPaddings()
153 void* buffer = tensor->GetBuffer(); in SetPadModeOrPaddings()
[all …]
H A Dfullconnection_builder.cpp59 if (tensor->GetDataType() != OH_NN_BOOL) { in SetHasBias()
69 void* buffer = tensor->GetBuffer(); in SetHasBias()
81 if (tensor->GetDataType() != OH_NN_BOOL) { in SetUseAxis()
91 void* buffer = tensor->GetBuffer(); in SetUseAxis()
110 tensor->IdentifyOpParameter(); in SetFullConnectionActivation()
117 if (tensor->GetDataType() != OH_NN_INT8) { in SetFullConnectionActivation()
122 void* buffer = tensor->GetBuffer(); in SetFullConnectionActivation()
140 tensor->IdentifyOpParameter(); in SetAxis()
147 if (tensor->GetDataType() != OH_NN_INT64) { in SetAxis()
152 void* buffer = tensor->GetBuffer(); in SetAxis()
[all …]
H A Dargmax_builder.cpp32 tensor->IdentifyOpParameter(); in SetAxis()
34 if (tensor->GetDataType() != OH_NN_INT64) { in SetAxis()
39 void* buffer = tensor->GetBuffer(); in SetAxis()
51 tensor->IdentifyOpParameter(); in SetTopK()
58 void* buffer = tensor->GetBuffer(); in SetTopK()
70 tensor->IdentifyOpParameter(); in SetKeepdims()
72 if (tensor->GetDataType() != OH_NN_BOOL) { in SetKeepdims()
77 void* buffer = tensor->GetBuffer(); in SetKeepdims()
89 tensor->IdentifyOpParameter(); in SetOutMaxValue()
96 void* buffer = tensor->GetBuffer(); in SetOutMaxValue()
[all …]
H A Dmatmul_builder.cpp37 tensor->IdentifyOpParameter(); in SetTransposeA()
38 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetTransposeA()
43 if (tensor->GetDataType() != OH_NN_BOOL) { in SetTransposeA()
48 void* buffer = tensor->GetBuffer(); in SetTransposeA()
60 tensor->IdentifyOpParameter(); in SetTransposeB()
66 if (tensor->GetDataType() != OH_NN_BOOL) { in SetTransposeB()
71 void* buffer = tensor->GetBuffer(); in SetTransposeB()
83 tensor->IdentifyOpParameter(); in SetActivationType()
89 if (tensor->GetDataType() != OH_NN_INT8) { in SetActivationType()
94 void* buffer = tensor->GetBuffer(); in SetActivationType()
[all …]
H A DreduceL2_builder.cpp36 if (tensor->GetDataType() != OH_NN_FLOAT32) { in SetCoeff()
41 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetCoeff()
46 void* buffer = tensor->GetBuffer(); in SetCoeff()
58 if (tensor->GetDataType() != OH_NN_BOOL) { in SetReduceToEnd()
63 tensor->IdentifyOpParameter(); in SetReduceToEnd()
64 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetReduceToEnd()
69 void* buffer = tensor->GetBuffer(); in SetReduceToEnd()
81 if (tensor->GetDataType() != OH_NN_BOOL) { in SetKeepDims()
86 tensor->IdentifyOpParameter(); in SetKeepDims()
92 void* buffer = tensor->GetBuffer(); in SetKeepDims()
[all …]
H A Dreduceall_builder.cpp36 if (tensor->GetDataType() != OH_NN_FLOAT32) { in SetCoeff()
41 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetCoeff()
46 void* buffer = tensor->GetBuffer(); in SetCoeff()
58 if (tensor->GetDataType() != OH_NN_BOOL) { in SetReduceToEnd()
63 tensor->IdentifyOpParameter(); in SetReduceToEnd()
64 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetReduceToEnd()
69 void* buffer = tensor->GetBuffer(); in SetReduceToEnd()
81 if (tensor->GetDataType() != OH_NN_BOOL) { in SetKeepDims()
86 tensor->IdentifyOpParameter(); in SetKeepDims()
92 void* buffer = tensor->GetBuffer(); in SetKeepDims()
[all …]
H A Dreducemax_builder.cpp36 if (tensor->GetDataType() != OH_NN_FLOAT32) { in SetCoeff()
41 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetCoeff()
46 void* buffer = tensor->GetBuffer(); in SetCoeff()
58 if (tensor->GetDataType() != OH_NN_BOOL) { in SetReduceToEnd()
63 tensor->IdentifyOpParameter(); in SetReduceToEnd()
64 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetReduceToEnd()
69 void* buffer = tensor->GetBuffer(); in SetReduceToEnd()
81 if (tensor->GetDataType() != OH_NN_BOOL) { in SetKeepDims()
86 tensor->IdentifyOpParameter(); in SetKeepDims()
92 void* buffer = tensor->GetBuffer(); in SetKeepDims()
[all …]
H A Dreducemean_builder.cpp36 if (tensor->GetDataType() != OH_NN_FLOAT32) { in SetCoeff()
41 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetCoeff()
46 void* buffer = tensor->GetBuffer(); in SetCoeff()
58 if (tensor->GetDataType() != OH_NN_BOOL) { in SetReduceToEnd()
63 tensor->IdentifyOpParameter(); in SetReduceToEnd()
64 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetReduceToEnd()
69 void* buffer = tensor->GetBuffer(); in SetReduceToEnd()
81 tensor->IdentifyOpParameter(); in SetKeepDims()
87 if (tensor->GetDataType() != OH_NN_BOOL) { in SetKeepDims()
92 void* buffer = tensor->GetBuffer(); in SetKeepDims()
[all …]
H A Dreducemin_builder.cpp36 if (tensor->GetDataType() != OH_NN_FLOAT32) { in SetCoeff()
41 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetCoeff()
46 void* buffer = tensor->GetBuffer(); in SetCoeff()
58 if (tensor->GetDataType() != OH_NN_BOOL) { in SetReduceToEnd()
63 tensor->IdentifyOpParameter(); in SetReduceToEnd()
64 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetReduceToEnd()
69 void* buffer = tensor->GetBuffer(); in SetReduceToEnd()
81 if (tensor->GetDataType() != OH_NN_BOOL) { in SetKeepDims()
86 tensor->IdentifyOpParameter(); in SetKeepDims()
92 void* buffer = tensor->GetBuffer(); in SetKeepDims()
[all …]
H A Dreduceprod_builder.cpp36 if (tensor->GetDataType() != OH_NN_FLOAT32) { in SetCoeff()
41 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetCoeff()
46 void* buffer = tensor->GetBuffer(); in SetCoeff()
58 if (tensor->GetDataType() != OH_NN_BOOL) { in SetReduceToEnd()
63 tensor->IdentifyOpParameter(); in SetReduceToEnd()
64 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetReduceToEnd()
69 void* buffer = tensor->GetBuffer(); in SetReduceToEnd()
81 tensor->IdentifyOpParameter(); in SetKeepDims()
87 if (tensor->GetDataType() != OH_NN_BOOL) { in SetKeepDims()
92 void* buffer = tensor->GetBuffer(); in SetKeepDims()
[all …]
H A Dreducesum_builder.cpp36 if (tensor->GetDataType() != OH_NN_FLOAT32) { in SetCoeff()
41 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetCoeff()
46 void* buffer = tensor->GetBuffer(); in SetCoeff()
58 if (tensor->GetDataType() != OH_NN_BOOL) { in SetReduceToEnd()
63 tensor->IdentifyOpParameter(); in SetReduceToEnd()
64 if (tensor->GetElementCount() != SCALE_LENGTH) { in SetReduceToEnd()
69 void* buffer = tensor->GetBuffer(); in SetReduceToEnd()
81 if (tensor->GetDataType() != OH_NN_BOOL) { in SetKeepDims()
86 tensor->IdentifyOpParameter(); in SetKeepDims()
92 void* buffer = tensor->GetBuffer(); in SetKeepDims()
[all …]
H A Dexp_builder.cpp33 if (tensor->GetDataType() != OH_NN_FLOAT32) { in SetBase()
38 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetBase()
43 void* buffer = tensor->GetBuffer(); in SetBase()
55 if (tensor->GetDataType() != OH_NN_FLOAT32) { in SetScale()
60 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetScale()
65 void* buffer = tensor->GetBuffer(); in SetScale()
77 if (tensor->GetDataType() != OH_NN_FLOAT32) { in SetShift()
82 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetShift()
87 void* buffer = tensor->GetBuffer(); in SetShift()
124 tensor->IdentifyOpParameter(); in Build()
[all …]
H A Dstrided_slice_builder.cpp52 if (tensor->GetDataType() != OH_NN_INT64) { in SetBeginMask()
57 void* buffer = tensor->GetBuffer(); in SetBeginMask()
69 if (tensor->GetDataType() != OH_NN_INT64) { in SetEndMask()
74 void* buffer = tensor->GetBuffer(); in SetEndMask()
86 if (tensor->GetDataType() != OH_NN_INT64) { in SetEllipsisMask()
91 void* buffer = tensor->GetBuffer(); in SetEllipsisMask()
103 if (tensor->GetDataType() != OH_NN_INT64) { in SetNewAxisMask()
108 void* buffer = tensor->GetBuffer(); in SetNewAxisMask()
125 void* buffer = tensor->GetBuffer(); in SetShrinkAxisMask()
165 tensor->IdentifyOpParameter(); in Build()
[all …]
H A Drange_builder.cpp33 if (tensor->GetDataType() != OH_NN_INT64) { in SetStart()
38 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetStart()
43 void* buffer = tensor->GetBuffer(); in SetStart()
55 if (tensor->GetDataType() != OH_NN_INT64) { in SetLimit()
60 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetLimit()
65 void* buffer = tensor->GetBuffer(); in SetLimit()
77 if (tensor->GetDataType() != OH_NN_INT64) { in SetDelta()
82 if (tensor->GetElementCount() != SCALAR_LENGTH) { in SetDelta()
87 void* buffer = tensor->GetBuffer(); in SetDelta()
124 tensor->IdentifyOpParameter(); in Build()
[all …]
/ohos5.0/docs/en/application-dev/reference/apis-mindspore-lite-kit/
H A Dtensor_8h.md1 # tensor.h
6 Provides tensor-related APIs, which can be used to create tensors and modify tensor information. Th…
33 …roy](_mind_spore.md#oh_ai_tensordestroy) (OH_AI_TensorHandle \*tensor) | Destroys a tensor object.|
34 …H_AI_TensorClone](_mind_spore.md#oh_ai_tensorclone) (OH_AI_TensorHandle tensor) | Clones a tensor.|
35 …ind_spore.md#oh_ai_tensorsetname) (OH_AI_TensorHandle tensor, const char \*name) | Sets the tensor…
36 …](_mind_spore.md#oh_ai_tensorgetname) (const OH_AI_TensorHandle tensor) | Obtains the tensor name.|
38 …ind_spore.md#oh_ai_tensorgetdatatype) (const OH_AI_TensorHandle tensor) | Obtains the tensor type.|
40 ….md#oh_ai_tensorgetshape) (const OH_AI_TensorHandle tensor, size_t \*shape_num) | Obtains the tens…
42 …ind_spore.md#oh_ai_tensorgetformat) (const OH_AI_TensorHandle tensor) | Obtains the tensor data fo…
43 …(_mind_spore.md#oh_ai_tensorsetdata) (OH_AI_TensorHandle tensor, void \*data) | Sets the tensor da…
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/inner_model/
H A Dnn_tensor_test.cpp53 OH_NN_Tensor tensor; variable
54 tensor.dataType = OH_NN_FLOAT32;
55 tensor.dimensionCount = 2;
56 tensor.dimensions = nullptr;
57 tensor.quantParam = nullptr;
58 tensor.type = OH_NN_TENSOR;
442 OH_NN_Tensor tensor; variable
443 tensor.dataType = OH_NN_FLOAT32;
444 tensor.dimensionCount = 2;
445 tensor.dimensions = dim;
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v2_0/inner_model/
H A Dnn_tensor_test.cpp53 OH_NN_Tensor tensor; variable
54 tensor.dataType = OH_NN_FLOAT32;
55 tensor.dimensionCount = 2;
56 tensor.dimensions = nullptr;
57 tensor.quantParam = nullptr;
58 tensor.type = OH_NN_TENSOR;
442 OH_NN_Tensor tensor; variable
443 tensor.dataType = OH_NN_FLOAT32;
444 tensor.dimensionCount = 2;
445 tensor.dimensions = dim;
[all …]
/ohos5.0/docs/zh-cn/application-dev/reference/apis-mindspore-lite-kit/
H A Dtensor_8h.md1 # tensor.h
33 | [OH_AI_TensorDestroy](_mind_spore.md#oh_ai_tensordestroy) (OH_AI_TensorHandle \*tensor) | 释放张量对象。…
34 | [OH_AI_TensorClone](_mind_spore.md#oh_ai_tensorclone) (OH_AI_TensorHandle tensor) | 深拷贝一个张量。 |
35 | [OH_AI_TensorSetName](_mind_spore.md#oh_ai_tensorsetname) (OH_AI_TensorHandle tensor, const char …
36 | [OH_AI_TensorGetName](_mind_spore.md#oh_ai_tensorgetname) (const OH_AI_TensorHandle tensor) | 获取张…
38 …rGetDataType](_mind_spore.md#oh_ai_tensorgetdatatype) (const OH_AI_TensorHandle tensor) | 获取张量类型。 |
42 | [OH_AI_TensorGetFormat](_mind_spore.md#oh_ai_tensorgetformat) (const OH_AI_TensorHandle tensor) |…
43 | [OH_AI_TensorSetData](_mind_spore.md#oh_ai_tensorsetdata) (OH_AI_TensorHandle tensor, void \*data…
44 | [OH_AI_TensorGetData](_mind_spore.md#oh_ai_tensorgetdata) (const OH_AI_TensorHandle tensor) | 获取张…
46 …ementNum](_mind_spore.md#oh_ai_tensorgetelementnum) (const OH_AI_TensorHandle tensor) | 获取张量元素数量。 |
[all …]

12345678910>>...14