SpaceToDepth - 1 vs 13

Files changed (1) hide show
  1. SpaceToDepth1 → SpaceToDepth13 +1 -0
SpaceToDepth1 → SpaceToDepth13 RENAMED
@@ -1 +1 @@
1
1
  SpaceToDepth rearranges blocks of spatial data into depth. More specifically,
2
2
  this op outputs a copy of the input tensor where values from the height and width dimensions
3
3
  are moved to the depth dimension.
4
4
  **Attributes**
5
5
  * **blocksize** (required):
6
6
  Blocks of [blocksize, blocksize] are moved.
7
7
  **Inputs**
8
8
  * **input** (heterogeneous) - **T**:
9
9
  Input tensor of [N,C,H,W], where N is the batch axis, C is the
10
10
  channel or depth, H is the height and W is the width.
11
11
  **Outputs**
12
12
  * **output** (heterogeneous) - **T**:
13
13
  Output tensor of [N, C * blocksize * blocksize, H/blocksize,
14
14
  W/blocksize].
15
15
  **Type Constraints**
16
16
  * **T** in (
17
+ tensor(bfloat16),
17
18
  tensor(bool),
18
19
  tensor(complex128),
19
20
  tensor(complex64),
20
21
  tensor(double),
21
22
  tensor(float),
22
23
  tensor(float16),
23
24
  tensor(int16),
24
25
  tensor(int32),
25
26
  tensor(int64),
26
27
  tensor(int8),
27
28
  tensor(string),
28
29
  tensor(uint16),
29
30
  tensor(uint32),
30
31
  tensor(uint64),
31
32
  tensor(uint8)
32
33
  ):
33
34
  Constrain input and output types to all tensor types.