Home
last modified time | relevance | path

Searched refs:keepDims (Results 1 – 23 of 23) sorted by relevance

/ohos5.0/docs/zh-cn/device-dev/reference/hdi-apis/nnrt/
H A D_reduce_fusion_v10.md20 - output,执行Reduce之后的m维的tensor,其数据类型和x相同。当keepDims为false时,m<n;当keepDims为true时,m==n。
34 | boolean [keepDims](#keepdims) | 维度是否保持不变。 |
53 ### keepDims subsection
56 boolean ReduceFusion::keepDims
H A D_reduce_fusion_v20.md20 - output,执行Reduce之后的m维的tensor,其数据类型和x相同。当keepDims为false时,m<n;当keepDims为true时,m==n。
34 | boolean [keepDims](#keepdims) | 维度是否保持不变。 |
53 ### keepDims subsection
56 boolean ReduceFusion::keepDims
H A D_arg_max_fusion_v10.md32 | boolean [keepDims](#keepdims) | 是否保持输出维度和输入的维度是否相同。 |
49 ### keepDims subsection
52 boolean ArgMaxFusion::keepDims
H A D_arg_max_fusion_v20.md32 | boolean [keepDims](#keepdims) | 是否保持输出维度和输入的维度是否相同。 |
49 ### keepDims subsection
52 boolean ArgMaxFusion::keepDims
/ohos5.0/docs/zh-cn/application-dev/reference/apis-neural-network-runtime-kit/
H A D_neural_network_runtime.md617keepDims为false时,计算指定维度上的平均值,减少input的维数;当keepDims为true时,计算指定维度上的平均值,保留相应的维度。<br/>输入:<br/>- input,n维…
625keepDims,布尔值,是否保留维度的标志位。当keepDims为true时,output的维数和input保持一致;当keepDims为false时, output的维数缩减。<br/>- r…
626keepDims为false时,减少input的维数;当keepDims为true时,保留相应的维度。<br/>输入:<br/>- n维输入张量,n&lt;8。<br/>- 一维张量,指定计算逻辑…
674keepDims是false,输出张量的维度将会缩减; 如果keepDims为true,输出张量的维度和输入张量保持不变。<br/>输入:<br/>- input,n维张量,其中n&lt;8。<b…
675keepDims是false,输出张量的维度将会缩减; 如果keepDims为true,输出张量的维度和输入张量保持不变。<br/>输入:<br/>- input,n维张量,其中n&lt;8。<b…
676keepDims是false,输出张量的维度将会缩减; 如果keepDims为true,输出张量的维度和输入张量保持不变。<br/>输入:<br/>- input,n维张量,其中n&lt;8。<b…
681keepDims是false,输出张量的维度将会缩减; 如果keepDims为true,输出张量的维度和输入张量保持不变。<br/>输入:<br/>- input,n维张量,其中n&lt;8。<b…
834 | OH_NN_REDUCE_MEAN_KEEP_DIMS | 当张量作为ReduceMean算子的keepDims参数时,使用本枚举值。 |
845 | OH_NN_REDUCE_ALL_KEEP_DIMS | 当张量作为ReduceAll算子的keepDims参数时,使用本枚举值。 |
850 | OH_NN_ARG_MAX_KEEPDIMS | 当张量作为ArgMax算子的keepDims参数时,使用本枚举值。 |
[all …]
H A D_neural_nework_runtime.md617keepDims为false时,计算指定维度上的平均值,减少input的维数;当keepDims为true时,计算指定维度上的平均值,保留相应的维度。<br/>输入:<br/>- input,n维…
625keepDims,布尔值,是否保留维度的标志位。当keepDims为true时,output的维数和input保持一致;当keepDims为false时, output的维数缩减。<br/>- r…
626keepDims为false时,减少input的维数;当keepDims为true时,保留相应的维度。<br/>输入:<br/>- n维输入张量,n&lt;8。<br/>- 一维张量,指定计算逻辑…
674keepDims是false,输出张量的维度将会缩减; 如果keepDims为true,输出张量的维度和输入张量保持不变。<br/>输入:<br/>- input,n维张量,其中n&lt;8。<b…
675keepDims是false,输出张量的维度将会缩减; 如果keepDims为true,输出张量的维度和输入张量保持不变。<br/>输入:<br/>- input,n维张量,其中n&lt;8。<b…
676keepDims是false,输出张量的维度将会缩减; 如果keepDims为true,输出张量的维度和输入张量保持不变。<br/>输入:<br/>- input,n维张量,其中n&lt;8。<b…
681keepDims是false,输出张量的维度将会缩减; 如果keepDims为true,输出张量的维度和输入张量保持不变。<br/>输入:<br/>- input,n维张量,其中n&lt;8。<b…
834 | OH_NN_REDUCE_MEAN_KEEP_DIMS | 当张量作为ReduceMean算子的keepDims参数时,使用本枚举值。 |
845 | OH_NN_REDUCE_ALL_KEEP_DIMS | 当张量作为ReduceAll算子的keepDims参数时,使用本枚举值。 |
850 | OH_NN_ARG_MAX_KEEPDIMS | 当张量作为ArgMax算子的keepDims参数时,使用本枚举值。 |
[all …]
/ohos5.0/drivers/interface/nnrt/v1_0/
H A DNodeAttrTypes.idl38 boolean keepDims;
205 boolean keepDims;
/ohos5.0/drivers/interface/nnrt/v2_0/
H A DNodeAttrTypes.idl38 boolean keepDims;
206 boolean keepDims;
/ohos5.0/foundation/ability/idl_tool/test/hdi_unittest/nnrt_v1_0/foo/nnrt/v1_0/
H A DNodeAttrTypes.idl38 boolean keepDims;
205 boolean keepDims;
/ohos5.0/foundation/ability/idl_tool/test/hdi_unittest/nnrt_v2_0/foo/nnrt/v2_0/
H A DNodeAttrTypes.idl38 boolean keepDims;
206 boolean keepDims;
/ohos5.0/docs/en/application-dev/reference/apis-neural-network-runtime-kit/
H A D_neural_network_runtime.md676keepDims** is **false**, the dimension of the output tensor is reduced. If **keepDims** is **true*…
681keepDims** is **false**, the dimension of the output tensor is reduced. If **keepDims** is **true*…
834 | OH_NN_REDUCE_MEAN_KEEP_DIMS | Used when the tensor is used as the **keepDims** parameter of the R…
844 | OH_NN_REDUCE_PROD_KEEP_DIMS | Used when the tensor is used as the **keepDims** parameter of the R…
845 | OH_NN_REDUCE_ALL_KEEP_DIMS | Used when the tensor is used as the **keepDims** parameter of the Ar…
850 | OH_NN_ARG_MAX_KEEPDIMS | Used when the tensor is used as the **keepDims** parameter of the ArgMax…
879 | OH_NN_ALL_KEEP_DIMS<sup>12+</sup> | Used when the tensor is used as the **keepDims** parameter of…
925 | OH_NN_REDUCE_MAX_KEEP_DIMS<sup>12+</sup> | Used when the tensor is used as the **keepDims** param…
928 | OH_NN_REDUCE_MIN_KEEP_DIMS<sup>12+</sup> | Used when the tensor is used as the **keepDims** param…
931 | OH_NN_REDUCE_SUM_KEEP_DIMS<sup>12+</sup> | Used when the tensor is used as the **keepDims** param…
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/tflite/delegates/nnrt_delegate/
H A Dnnrt_op_builder.cpp225 const int32_t keepDims = (builtin->keep_dims); in AddReduceMeanParams() local
226 mappingArgs.builder->AddScalarBoolTensor(keepDims, OH_NN_REDUCE_MEAN_KEEP_DIMS); in AddReduceMeanParams()
/ohos5.0/foundation/ability/idl_tool/test/hdi_unittest/nnrt_v1_0/dump_ast_target/
H A Ddump.txt79 boolean keepDims;
220 boolean keepDims;
/ohos5.0/foundation/ability/idl_tool/test/hdi_unittest/nnrt_v2_0/dump_ast_target/
H A Ddump.txt144 boolean keepDims;
286 boolean keepDims;
/ohos5.0/foundation/ability/idl_tool/test/hdi_unittest/nnrt_v1_0/cpp_target/nnrt/v1_0/
H A Dnode_attr_types.h.txt71 bool keepDims;
212 bool keepDims;
/ohos5.0/foundation/ability/idl_tool/test/hdi_unittest/nnrt_v2_0/cpp_target/nnrt/v2_0/
H A Dnode_attr_types.h.txt71 bool keepDims;
213 bool keepDims;
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/device_manager/
H A Ddevice_manager_test.cpp213 int64_t keepDims {0}; variable
214 void* primitive = mindspore::lite::MindIR_All_CreatePrimitive(keepDims);
234 bool keepDims {false}; variable
236 …void* primitive = mindspore::lite::MindIR_ArgMaxFusion_CreatePrimitive(axis, topK, keepDims, outMa…
1484 bool keepDims {false}; variable
1485 …void* primitive = mindspore::lite::MindIR_ReduceFusion_CreatePrimitive(keepDims, mode, reduceToEnd…
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/inner_model/
H A Dinner_model_test.cpp1131 int64_t keepDims {0}; variable
1132 void* primitive = mindspore::lite::MindIR_All_CreatePrimitive(keepDims);
1152 bool keepDims {false}; variable
1154 …void* primitive = mindspore::lite::MindIR_ArgMaxFusion_CreatePrimitive(axis, topK, keepDims, outMa…
2402 bool keepDims {false}; variable
2403 …void* primitive = mindspore::lite::MindIR_ReduceFusion_CreatePrimitive(keepDims, mode, reduceToEnd…
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v2_0/inner_model/
H A Dinner_model_test.cpp1070 int64_t keepDims {0}; variable
1071 void* primitive = mindspore::lite::MindIR_All_CreatePrimitive(keepDims);
1091 bool keepDims {false}; variable
1093 …void* primitive = mindspore::lite::MindIR_ArgMaxFusion_CreatePrimitive(axis, topK, keepDims, outMa…
2341 bool keepDims {false}; variable
2342 …void* primitive = mindspore::lite::MindIR_ReduceFusion_CreatePrimitive(keepDims, mode, reduceToEnd…
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dlite_graph_to_hdi_model_v1_0.cpp82 argMaxFusion.keepDims = mindspore::lite::MindIR_ArgMaxFusion_GetKeepDims(primitive); in ConvertArgMaxFusion()
553 reduceFusion.keepDims = mindspore::lite::MindIR_ReduceFusion_GetKeepDims(primitive); in ConvertReduceFusion()
H A Dlite_graph_to_hdi_model_v2_0.cpp82 argMaxFusion.keepDims = mindspore::lite::MindIR_ArgMaxFusion_GetKeepDims(primitive); in ConvertArgMaxFusion()
554 reduceFusion.keepDims = mindspore::lite::MindIR_ReduceFusion_GetKeepDims(primitive); in ConvertReduceFusion()
H A Dlite_graph_to_hdi_model_v2_1.cpp80 all.keepDims = mindspore::lite::MindIR_All_GetKeepDims(primitive); in ConvertAll()
99 argMaxFusion.keepDims = mindspore::lite::MindIR_ArgMaxFusion_GetKeepDims(primitive); in ConvertArgMaxFusion()
1134 reduceFusion.keepDims = mindspore::lite::MindIR_ReduceFusion_GetKeepDims(primitive); in ConvertReduceFusion()
/ohos5.0/drivers/interface/nnrt/v2_1/
H A DNodeAttrTypes.idl153 long keepDims;
179 boolean keepDims;
1836 …* * An m-dimensional tensor, with the same data type as <b>x</b>. If <b>keepDims</b> is <b>false</…
1837 * If <b>keepDims</b> is <b>true</b>, m==n.
1845 boolean keepDims;