Home
last modified time | relevance | path

Searched refs:sizeSplits (Results 1 – 22 of 22) sorted by relevance

/ohos5.0/docs/zh-cn/device-dev/reference/hdi-apis/nnrt/
H A D_split_v20.md31sizeSplits](#sizesplits) | 指定 输入的张量沿 axis 轴拆分后,每个张量的大小。 如果 sizeSplits 的数据为空,则 sizeSplits 被拆分成大小均等的…
58 ### sizeSplits subsection
61 long [] Split::sizeSplits
65 …is 轴拆分后,每个张量的大小。 如果 sizeSplits 的数据为空,则 sizeSplits 被拆分成大小均等的 张量,此时要求 x.shape[axis] 可以被 outputNum 整除…
H A D_split_v10.md31 | long[] [sizeSplits](#sizesplits) | 指定 输入的张量沿 axis 轴拆分后,每个张量的大小。 |
62 ### sizeSplits subsection
65 long [] Split::sizeSplits
73 - 如果 sizeSplits 的数据为空,则 sizeSplits 被拆分成大小均等的 张量,此时要求 x.shape[axis] 可以被 outputNum 整除;
75 - 如果 sizeSplits 不为空,则要求 sizeSplits 所有元素之和等于 x.shape[axis]。
/ohos5.0/drivers/interface/nnrt/v1_0/
H A DNodeAttrTypes.idl261 long[] sizeSplits;
/ohos5.0/drivers/interface/nnrt/v2_0/
H A DNodeAttrTypes.idl262 long[] sizeSplits;
/ohos5.0/foundation/ability/idl_tool/test/hdi_unittest/nnrt_v1_0/foo/nnrt/v1_0/
H A DNodeAttrTypes.idl261 long[] sizeSplits;
/ohos5.0/foundation/ability/idl_tool/test/hdi_unittest/nnrt_v2_0/foo/nnrt/v2_0/
H A DNodeAttrTypes.idl262 long[] sizeSplits;
/ohos5.0/foundation/ability/idl_tool/test/hdi_unittest/nnrt_v1_0/cpp_target/nnrt/v1_0/
H A Dnode_attr_types.cpp.txt1273 if (!WritePodArray(data, dataBlock.sizeSplits)) {
1274 HDF_LOGE("%{public}s: failed to write dataBlock.sizeSplits", __func__);
1292 if (!ReadPodArray(data, dataBlock.sizeSplits)) {
1293 HDF_LOGE("%{public}s: failed to read dataBlock.sizeSplits", __func__);
H A Dnode_attr_types.h.txt259 std::vector<int64_t> sizeSplits;
/ohos5.0/foundation/ability/idl_tool/test/hdi_unittest/nnrt_v2_0/cpp_target/nnrt/v2_0/
H A Dnode_attr_types.cpp.txt1287 if (!WritePodArray(data, dataBlock.sizeSplits)) {
1288 HDF_LOGE("%{public}s: failed to write dataBlock.sizeSplits", __func__);
1306 if (!ReadPodArray(data, dataBlock.sizeSplits)) {
1307 HDF_LOGE("%{public}s: failed to read dataBlock.sizeSplits", __func__);
H A Dnode_attr_types.h.txt260 std::vector<int64_t> sizeSplits;
/ohos5.0/docs/zh-cn/application-dev/reference/apis-neural-network-runtime-kit/
H A D_neural_network_runtime.md607sizeSplits,一维张量,指定 张量 沿 axis 轴拆分后,每个 张量 的大小,sizeSplits 类型为 int。 如果 sizeSplits 的数据为空,则 张量 被拆分成大小均等的…
H A D_neural_nework_runtime.md607sizeSplits,一维张量,指定 张量 沿 axis 轴拆分后,每个 张量 的大小,sizeSplits 类型为 int。 如果 sizeSplits 的数据为空,则 张量 被拆分成大小均等的…
/ohos5.0/foundation/ability/idl_tool/test/hdi_unittest/nnrt_v1_0/dump_ast_target/
H A Ddump.txt267 long[] sizeSplits;
/ohos5.0/foundation/ability/idl_tool/test/hdi_unittest/nnrt_v2_0/dump_ast_target/
H A Ddump.txt333 long[] sizeSplits;
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dlite_graph_to_hdi_model_v1_0.cpp709 split.sizeSplits = mindspore::lite::MindIR_Split_GetSizeSplits(primitive); in ConvertSplit()
H A Dlite_graph_to_hdi_model_v2_0.cpp710 split.sizeSplits = mindspore::lite::MindIR_Split_GetSizeSplits(primitive); in ConvertSplit()
H A Dlite_graph_to_hdi_model_v2_1.cpp1372 split.sizeSplits = mindspore::lite::MindIR_Split_GetSizeSplits(primitive); in ConvertSplit()
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/device_manager/
H A Ddevice_manager_test.cpp1989 std::vector<int64_t> sizeSplits; variable
1991 void* primitive = mindspore::lite::MindIR_Split_CreatePrimitive(outputNum, sizeSplits, axis);
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/inner_model/
H A Dinner_model_test.cpp2907 std::vector<int64_t> sizeSplits; variable
2909 void* primitive = mindspore::lite::MindIR_Split_CreatePrimitive(outputNum, sizeSplits, axis);
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v2_0/inner_model/
H A Dinner_model_test.cpp2846 std::vector<int64_t> sizeSplits; variable
2848 void* primitive = mindspore::lite::MindIR_Split_CreatePrimitive(outputNum, sizeSplits, axis);
/ohos5.0/drivers/interface/nnrt/v2_1/
H A DNodeAttrTypes.idl2217 * If <b>sizeSplits</b> is empty, <b>x</b> will be evenly split into tensors of the same size.
2219 * If <b>sizeSplits</b> is not empty, the sum of all elements must be equal to x.shape[axis].
2221 long[] sizeSplits;
/ohos5.0/docs/en/application-dev/reference/apis-neural-network-runtime-kit/
H A D_neural_network_runtime.md607sizeSplits**: size of each tensor split from the input. The value is a 1D tensor of the **int** ty…