Home
last modified time | relevance | path

Searched refs:shape (Results 1 - 3 of 3) sorted by relevance

/interface/sdk_c/ai/neural_network_runtime/
H A Dneural_network_core.h375 * The {@link NN_TensorDesc} describes various tensor attributes, such as name/data type/shape/format, etc.\n
491 * @brief Sets the shape of a {@link NN_TensorDesc}.
493 * After the {@link NN_TensorDesc} instance is created, call this method to set the tensor shape. \n
495 * if <b>tensorDesc</b> or <b>shape</b> is a null pointer, or <b>shapeLength</b> is 0, this method will return error code. \n
498 * @param shape The shape list of the tensor that needs to be set.
499 * @param shapeLength The length of the shape list that needs to be set.
501 * {@link OH_NN_SUCCESS} set tensor shape successfully.\n
502 * {@link OH_NN_INVALID_PARAMETER} fail to set tensor shape. The possible reason for failure
503 * is that the <b>tensorDesc</b> or <b>shape</
[all...]
/interface/sdk_c/third_party/mindspore/kits/
H A Dtensor.h61 * @param shape The shape of the tensor.
62 * @param shape_num The num of the shape.
68 OH_AI_API OH_AI_TensorHandle OH_AI_TensorCreate(const char *name, OH_AI_DataType type, const int64_t *shape,
119 * @brief Set the shape for the tensor.
121 * @param shape The shape array.
122 * @param shape_num Dimension of shape.
125 OH_AI_API void OH_AI_TensorSetShape(OH_AI_TensorHandle tensor, const int64_t *shape, size_t shape_num);
128 * @brief Obtain the shape o
[all...]
H A Dmodel.h58 int64_t shape[OH_AI_MAX_SHAPE_NUM]; member
389 * @param changeable_weights_name The set the name of these weight tensors, whose shape is changeable.

Completed in 2 milliseconds