Searched refs:evalData (Results 1 – 1 of 1) sorted by relevance
159 int getFlatIndex(int index, const EvalData<T>& evalData) { in getFlatIndex() argument161 for (int i = 0, nD = evalData.numDims; i < nD; ++i) { in getFlatIndex()163 getPadding(evalData.paddingTensor, i, &leftPad, &rightPad); in getFlatIndex()164 const int dimensionIndex = index / evalData.outputDimsNumElements[i]; in getFlatIndex()168 int(getSizeOfDimension(evalData.inputTensorShape, i)), evalData.offset); in getFlatIndex()169 flatIndex += indexInInput * evalData.inputDimsNumElements[i]; in getFlatIndex()170 index %= evalData.outputDimsNumElements[i]; in getFlatIndex()176 void run(const EvalData<T>& evalData, const int outputSize) { in run() argument178 const auto* inputData = evalData.inputData; in run()179 auto* outputData = evalData.outputData; in run()[all …]