1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
use tensor_rs::tensor::Tensor;
use super::{OpTrait, OpHandle};
pub struct MaxPool2d {
handle: OpHandle,
kernel_size: (usize, usize),
stride: (usize, usize),
padding: Tensor,
dilation: (usize, usize),
return_indices: bool,
ceil_mode: bool,
}
impl MaxPool2d {
pub fn new(kernel_size: Option<(usize, usize)>,
stride: Option<(usize, usize)>,
padding: Option<Tensor>,
dilation: Option<(usize, usize)>,
return_indices: Option<bool>,
ceil_mode: Option<bool>,) -> MaxPool2d {
let kernel_size = if let Some(v) = kernel_size {v} else {(2, 2)};
let stride = if let Some(v) = stride {v} else {(2, 2)};
let padding = if let Some(v) = padding {v} else {Tensor::zeros(&[1])};
let dilation = if let Some(v) = dilation {v} else {(2, 2)};
let return_indices = if let Some(v) = return_indices {v} else {false};
let ceil_mode = if let Some(v) = ceil_mode {v} else {false};
MaxPool2d {
handle: OpHandle::new(),
kernel_size,
stride,
padding,
dilation,
return_indices,
ceil_mode,
}
}
fn get_handle(&self) -> &OpHandle {
&self.handle
}
fn get_handle_mut(&mut self) -> &mut OpHandle {
&mut self.handle
}
}
impl OpTrait for MaxPool2d {
fn get_name(&self) -> String {
"max_pool2d".to_string()
}
fn get_input_size(&self) -> usize {
1
}
fn get_output_size(&self) -> usize {
1
}
fn apply(&self, input: &[Tensor], output: &[Tensor]) {
unimplemented!();
}
fn grad(&self, input: &[Tensor], output_grad: &[Tensor], input_grad: &[Tensor]) {
unimplemented!();
}
fn get_values(&self) -> Vec<Tensor> {
Vec::new()
}
fn get_grads(&self) -> Vec<Tensor> {
Vec::new()
}
fn set_values(&self, _v: &[Tensor]) {
}
}
