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
Loading
Pipeline #64480 failed
...@@ -105,18 +105,18 @@ Aidge::Conv_Op<DIM>::computeReceptiveField( ...@@ -105,18 +105,18 @@ Aidge::Conv_Op<DIM>::computeReceptiveField(
std::vector<DimSize_t> inputDims{outputDims[0], getInput(0)->dims()[1]}; std::vector<DimSize_t> inputDims{outputDims[0], getInput(0)->dims()[1]};
for (DimIdx_t i = 0; i < DIM; ++i) { for (DimIdx_t i = 0; i < DIM; ++i) {
inputDims.push_back((outputDims[2+static_cast<std::size_t>(i)] - 1) 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 + 1
+ (mAttributes->template getAttr<ConvAttr::KernelDims>()[static_cast<std::size_t>(i)] - 1) + (kernelDims()[static_cast<std::size_t>(i)] - 1)
* mAttributes->template getAttr<ConvAttr::DilationDims>()[static_cast<std::size_t>(i)]); * dilationDims()[static_cast<std::size_t>(i)]);
inputIdxDims[2+i] *= mAttributes->template getAttr<ConvAttr::StrideDims>()[static_cast<std::size_t>(i)]; inputIdxDims[2+i] *= strideDims()[static_cast<std::size_t>(i)];
} }
// Weight // Weight
// same output value, every input channel is used // same output value, every input channel is used
std::vector<DimSize_t> weightDims{outputDims[1], getInput(0)->dims()[1]}; std::vector<DimSize_t> weightDims{outputDims[1], getInput(0)->dims()[1]};
for (std::size_t i = 0; i < DIM; ++i) { 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); std::vector<DimSize_t> weightIdxDims = std::vector<DimSize_t>(DIM+2, 0);
weightIdxDims[0] = firstEltDims[1]; 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