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_r366605962
 
 

 ##########
 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.
 
 Review comment:
   ```suggestion
    * \param out_shape_data returning array of pointers to head of the output 
shape.
   ```

----------------------------------------------------------------
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

Reply via email to