Skip to content

Commit

Permalink
[Layer] Enable mixed precision - pooling2d_layer
Browse files Browse the repository at this point in the history 
Enable Mixed precision on Pooling 2D Layer
- I modified it to properly cast for the case of FP16 so that the mixed precision function can be activated on the existing pooling 2d layer.

**Self evaluation:**
1. Build test:	 [X]Passed [ ]Failed [ ]Skipped
2. Run test:	 [X]Passed [ ]Failed [ ]Skipped

Signed-off-by: Donghak PARK <[email protected]>
  • Loading branch information
@DonghakPark
DonghakPark committed Jun 5, 2024
1 parent 8fce360 commit 1d9af0d
Show file tree
Hide file tree
Showing 2 changed files with 208 additions and 117 deletions.
Loading

0 comments on commit 1d9af0d

Please sign in to comment.