1
2
3
4
5
6
7
8
|
HloModule convolution.167:
ENTRY %convolution.167 (parameter.0: f32[16,28,28,128], parameter.1: f32[3,3,128,128]) -> f32[16,28,28,128] {
%parameter.0 = f32[16,28,28,128]{3,0,2,1} parameter(0)
%parameter.1 = f32[3,3,128,128]{3,2,1,0} parameter(1)
ROOT %convolution.167 = f32[16,28,28,128]{3,0,2,1} convolution(f32[16,28,28,128]{3,0,2,1} %parameter.0, f32[3,3,128,128]{3,2,1,0} %parameter.1), window={size=3x3 pad=1_1x1_1}, dim_labels=b01f_01oi->b01f
}
|
