Lines Matching refs:inputShape
41 void transposeFirstTwoDims(const T* input, const Shape& inputShape, T* output) { in transposeFirstTwoDims() argument
42 const uint32_t firstDimSize = getSizeOfDimension(inputShape, 0); in transposeFirstTwoDims()
43 const uint32_t secondDimSize = getSizeOfDimension(inputShape, 1); in transposeFirstTwoDims()
44 const uint32_t inputSize = getSizeOfDimension(inputShape, 2); in transposeFirstTwoDims()
59 Shape inputShape = context->getInputShape(kInputTensor); in executeTyped() local
79 inputTransposed.resize(getNumberOfElements(inputShape)); in executeTyped()
81 transposeFirstTwoDims(input, inputShape, inputTransposed.data()); in executeTyped()
84 std::swap(inputShape.dimensions[0], inputShape.dimensions[1]); in executeTyped()
88 const uint32_t maxTime = getSizeOfDimension(inputShape, 0); in executeTyped()
89 const uint32_t batchSize = getSizeOfDimension(inputShape, 1); in executeTyped()
90 const uint32_t inputSize = getSizeOfDimension(inputShape, 2); in executeTyped()
94 Shape fixedTimeInputShape = inputShape; in executeTyped()
96 fixedTimeInputShape.dimensions[0] = inputShape.dimensions[1]; in executeTyped()
97 fixedTimeInputShape.dimensions[1] = inputShape.dimensions[2]; in executeTyped()