Skip to content
Permalink
Browse files

minor proto sp fixes

  • Loading branch information...
brettkoonce committed Feb 11, 2018
1 parent 3c30189 commit 6d780e67367ec721752f4ab7a365c4f803e2dd9d
Showing with 5 additions and 5 deletions.
  1. +2 −2 mlmodel/format/DictVectorizer.proto
  2. +3 −3 mlmodel/format/NeuralNetwork.proto
@@ -16,12 +16,12 @@ package CoreML.Specification;
* The output array will be equal in length to the index mapping vector parameter.
* All keys in the input dictionary must be present in the index mapping vector.
*
* For each item in the input dictionary, insert its value in the ouput array.
* For each item in the input dictionary, insert its value in the output array.
* The position of the insertion is determined by the position of the item's key
* in the index mapping. Any keys not present in the input dictionary, will be
* zero in the output array.
*
* For example: if the ``stringToIndex`` paramter is set to ``["a", "c", "b", "z"]``,
* For example: if the ``stringToIndex`` parameter is set to ``["a", "c", "b", "z"]``,
* then an input of ``{"a": 4, "c": 8}`` will produce an output of ``[4, 8, 0, 0]``.
*
*/
@@ -639,7 +639,7 @@ message ConvolutionLayerParams {
/**
* Group convolution, i.e. weight reuse along channel axis.
* Input and kernels are divided into g groups
* and convolution / deconvoltuion is applied within the groups independently.
* and convolution / deconvolution is applied within the groups independently.
* If not set or 0, it is set to the default value 1.
*/
uint64 nGroups = 10;
@@ -1532,7 +1532,7 @@ message PermuteLayerParams {
* mode == DEPTH_TO_SPACE
* ``[C_out, H_out, W_out]`` : ``[C/(blockSize * blockSize), H * blockSize, W * blockSize]``.
* Square of blockSize must divide C.
* Reverse of SPACE_TO_DEPTH. Data is moved from the channel dimension to the spatial dimenions.
* Reverse of SPACE_TO_DEPTH. Data is moved from the channel dimension to the spatial dimensions.
*
*/
message ReorganizeDataLayerParams {
@@ -1570,7 +1570,7 @@ message SliceLayerParams {

uint64 startIndex = 1; /// start of the sliced section. Inclusive.
int64 endIndex = 2; /// end of sliced section. Exclusive.
uint64 stride = 3; /// The step size. Must be postive.
uint64 stride = 3; /// The step size. Must be positive.

enum SliceAxis {
CHANNEL_AXIS = 0;

0 comments on commit 6d780e6

Please sign in to comment.
You can’t perform that action at this time.