Skip to content
Snippets Groups Projects
Commit 8dd047c2 authored by Grégoire Kubler's avatar Grégoire Kubler
Browse files

chore : [Conv] updated the method for attribute accession to ease code reading

parent 51461929
No related branches found
No related tags found
No related merge requests found
Pipeline #64480 failed
......@@ -105,18 +105,18 @@ Aidge::Conv_Op<DIM>::computeReceptiveField(
std::vector<DimSize_t> inputDims{outputDims[0], getInput(0)->dims()[1]};
for (DimIdx_t i = 0; i < DIM; ++i) {
inputDims.push_back((outputDims[2+static_cast<std::size_t>(i)] - 1)
* mAttributes->template getAttr<ConvAttr::StrideDims>()[static_cast<std::size_t>(i)]
* strideDims()[static_cast<std::size_t>(i)]
+ 1
+ (mAttributes->template getAttr<ConvAttr::KernelDims>()[static_cast<std::size_t>(i)] - 1)
* mAttributes->template getAttr<ConvAttr::DilationDims>()[static_cast<std::size_t>(i)]);
inputIdxDims[2+i] *= mAttributes->template getAttr<ConvAttr::StrideDims>()[static_cast<std::size_t>(i)];
+ (kernelDims()[static_cast<std::size_t>(i)] - 1)
* dilationDims()[static_cast<std::size_t>(i)]);
inputIdxDims[2+i] *= strideDims()[static_cast<std::size_t>(i)];
}
// Weight
// same output value, every input channel is used
std::vector<DimSize_t> weightDims{outputDims[1], getInput(0)->dims()[1]};
for (std::size_t i = 0; i < DIM; ++i) {
weightDims.push_back(mAttributes->template getAttr<ConvAttr::KernelDims>()[i]);
weightDims.push_back(kernelDims()[i]);
}
std::vector<DimSize_t> weightIdxDims = std::vector<DimSize_t>(DIM+2, 0);
weightIdxDims[0] = firstEltDims[1];
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment