Searched refs:INPUT_RANK (Results 1 – 1 of 1) sorted by relevance
34 static const int INPUT_RANK = 4; variable92 if (weightShape.size() != INPUT_RANK) { in SetKernelSize()221 if (inputShape.size() != INPUT_RANK) { in Build()