Lines Matching defs:inputShape

2128   auto inputShape = inputType.getShape();
2155     if (inputShape[inputDimIdx] != initShape[initDimIdx])
2158                            << ". input: " << inputShape[inputDimIdx]
2613   ArrayRef<int64_t> inputShape = inputType.getShape();
2615   if (failed(verifyCompatibleShape(inputShape, outputShape)))
2790   ArrayRef<int64_t> inputShape = inputType.getShape();
2791   int64_t inputRank = inputShape.size();
2813   ArrayRef<int64_t> inputShape = inputType.getShape();
2814   int64_t inputRank = inputShape.size();
2839   ArrayRef<int64_t> inputShape = inputType.getShape();
2840   int64_t inputRank = inputShape.size();
3057   ArrayRef<int64_t> inputShape = inputType.getShape();
3058   int64_t inputH = inputShape[getInputHDim()];
3059   int64_t inputW = inputShape[getInputWDim()];
3086   expectedOutputShape[getOutputNDim()] = inputShape[getInputNDim()];
3087   expectedOutputShape[getOutputCDim()] = inputShape[getInputCDim()];