Home
last modified time | relevance | path

Searched refs:inputChannels (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaDecomposeTransposeConv.cpp190 int64_t inputChannels = weightTy.getDimSize(3); in matchAndRewrite() local
224 stride[1], inputChannels}; in matchAndRewrite()
241 weightWidth / stride[1], inputChannels}; in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp1181 int32_t inputChannels = ShapedType::kDynamicSize; in inferReturnTypeComponents() local
1193 inputChannels = inputShape.getDimSize(3); in inferReturnTypeComponents()
1201 inputChannels = ShapedType::isDynamic(inputChannels) in inferReturnTypeComponents()
1203 : inputChannels; in inferReturnTypeComponents()
1209 if (!ShapedType::isDynamic(inputChannels) && in inferReturnTypeComponents()
1211 outputShape[3] = inputChannels * depthChannels; in inferReturnTypeComponents()