ChaiBapchya commented on a change in pull request #17309: adding docs for 64bit
C APIs of large tensor and removing not required int64 C APIs
URL: https://github.com/apache/incubator-mxnet/pull/17309#discussion_r366606053
##########
File path: include/mxnet/c_api.h
##########
@@ -1840,6 +1926,32 @@ MXNET_DLL int MXSymbolInferShapePartialEx(SymbolHandle
sym,
const int ***aux_shape_data,
int *complete);
+/*!
+ * \brief partially infer shape of unknown input shapes given the known one.
+ *
+ * Return partially inferred results if not all shapes could be inferred.
+ * The shapes are packed into a CSR matrix represented by arg_ind_ptr and
arg_shape_data
+ * The call will be treated as a kwargs call if key != nullptr or
num_args==0, otherwise it is positional.
+ * This api is available when MXNet is built with flag
+ * USE_INT64_TENSOR_SIZE=1 (not default) i.e. Large Tensor Support
+ *
+ * \param sym symbol handle
+ * \param num_args numbe of input arguments.
+ * \param keys the key of keyword args (optional)
+ * \param arg_ind_ptr the head pointer of the rows in CSR
+ * \param arg_shape_data the content of the CSR
+ * \param in_shape_size sizeof the returning array of in_shapes
+ * \param in_shape_ndim returning array of shape dimensions of eachs input
shape.
+ * \param in_shape_data returning array of pointers to head of the input shape.
+ * \param out_shape_size sizeof the returning array of out_shapes
+ * \param out_shape_ndim returning array of shape dimensions of eachs input
shape.
+ * \param out_shape_data returning array of pointers to head of the input
shape.
+ * \param aux_shape_size sizeof the returning array of aux_shapes
Review comment:
```suggestion
* \param aux_shape_size size of the returning array of aux_shapes
```
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
With regards,
Apache Git Services
