Skip to content

Commit

Permalink
[layer] enable mixed precision - reshape_layer
Browse files Browse the repository at this point in the history
enable mixed precision on reshape layer
- reshape layer only change dim, so change dimensions and check datatype

**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 committed Jun 5, 2024
1 parent a5b1545 commit 8fce360
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion nntrainer/layers/reshape_layer.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@ void ReshapeLayer::finalize(InitLayerContext &context) {
}

out_dim.batch(in_dim.batch());

out_dim.setDataType(context.getActivationDataType());
context.setOutputDimensions({out_dim});
}

Expand Down

0 comments on commit 8fce360

Please sign in to comment.