Searched refs:outputDimsNumElements (Results 1 – 1 of 1) sorted by relevance
114 outputDimsNumElements(std::move(theOutputDimsNumElements)), in EvalData()124 std::vector<int> outputDimsNumElements; member164 const int dimensionIndex = index / evalData.outputDimsNumElements[i]; in getFlatIndex()170 index %= evalData.outputDimsNumElements[i]; in getFlatIndex()206 std::vector<int> outputDimsNumElements(inputDims, 1); in eval() local209 outputDimsNumElements[i] = in eval()210 outputDimsNumElements[i + 1] * int(getSizeOfDimension(outputShape, i + 1)); in eval()221 const EvalData evalData(padding, inputShape, std::move(outputDimsNumElements), \ in eval()