Searched defs:Convolution2DAttributes (Results 1 – 1 of 1) sorted by relevance
267 struct Convolution2DAttributes { struct268 HW strides = HW(1, 1); // Along each axis.269 HW dilations = HW(1, 1); // Along each axis.270 Padding2D padding;272 Tensor<OHWI, DataType::FLOAT32> weights;273 Tensor<Linear, DataType::FLOAT32> bias; // optional275 int groups = 1; // optional, split channels dimension on equal groups