Lines Matching refs:inputShape
49 void transposeFirstTwoDims(const T* input, const Shape& inputShape, T* output) { in transposeFirstTwoDims() argument
50 const uint32_t firstDimSize = getSizeOfDimension(inputShape, 0); in transposeFirstTwoDims()
51 const uint32_t secondDimSize = getSizeOfDimension(inputShape, 1); in transposeFirstTwoDims()
52 const uint32_t inputSize = getSizeOfDimension(inputShape, 2); in transposeFirstTwoDims()
67 Shape inputShape = context->getInputShape(kInputTensor); in executeTyped() local
87 inputTransposed.resize(getNumberOfElements(inputShape)); in executeTyped()
89 transposeFirstTwoDims(input, inputShape, inputTransposed.data()); in executeTyped()
92 std::swap(inputShape.dimensions[0], inputShape.dimensions[1]); in executeTyped()
96 const uint32_t maxTime = getSizeOfDimension(inputShape, 0); in executeTyped()
97 const uint32_t batchSize = getSizeOfDimension(inputShape, 1); in executeTyped()
98 const uint32_t inputSize = getSizeOfDimension(inputShape, 2); in executeTyped()
102 Shape fixedTimeInputShape = inputShape; in executeTyped()
104 fixedTimeInputShape.dimensions[0] = inputShape.dimensions[1]; in executeTyped()
105 fixedTimeInputShape.dimensions[1] = inputShape.dimensions[2]; in executeTyped()