Skip to content

Commit 1e2b6a6

Browse files
committed
[OP] Update BiasAdd OP settings
1 parent 4d3dc00 commit 1e2b6a6

File tree

3 files changed

+4
-3
lines changed

3 files changed

+4
-3
lines changed

source/tnn/device/cpu/acc/cpu_bias_add_layer_acc.cc

+2-2
Original file line numberDiff line numberDiff line change
@@ -25,9 +25,9 @@ Status CpuBiasAddLayerAcc::Reshape(const std::vector<Blob *> &inputs, const std:
2525
}
2626

2727
Status CpuBiasAddLayerAcc::Forward(const std::vector<Blob *> &inputs, const std::vector<Blob *> &outputs) {
28-
auto resource = dynamic_cast<BiasAddLayerResource *>(resource_);
28+
auto resource = dynamic_cast<BatchNormLayerResource *>(resource_);
2929
if (!resource) {
30-
return Status(TNNERR_MODEL_ERR, "Error: BiasAddLayerResource is nil");
30+
return Status(TNNERR_MODEL_ERR, "Error: BatchNormLayerResource is nil");
3131
}
3232

3333
auto input_blob = inputs[0];

source/tnn/interpreter/tnn/layer_interpreter/bias_add_layer_interpreter.cc

+1-1
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ Status BiasAddLayerInterpreter::SaveProto(std::ostream& output_stream, LayerPara
3333
}
3434

3535
Status BiasAddLayerInterpreter::SaveResource(Serializer& serializer, LayerParam* param, LayerResource* resource) {
36-
CAST_OR_RET_ERROR(bias_res, BiasAddLayerResource, "invalid layer res to save", resource);
36+
CAST_OR_RET_ERROR(bias_res, BatchNormLayerResource, "invalid layer res to save", resource);
3737
serializer.PutRaw(bias_res->bias_handle);
3838
return TNN_OK;
3939
}

source/tnn/network/tensorrt/layer_builder/batch_norm_layer_builder.cc

+1
Original file line numberDiff line numberDiff line change
@@ -75,6 +75,7 @@ ILayer* BatchNormTRTLayerBuilder::AddToNetwork(INetworkDefinition* network) {
7575
}
7676

7777
REGISTER_TENSORRT_LAYER_BUILDER(BatchNorm, LAYER_BATCH_NORM);
78+
REGISTER_TENSORRT_LAYER_BUILDER(BatchNorm, LAYER_BIAS_ADD);
7879
REGISTER_TENSORRT_LAYER_BUILDER(BatchNorm, LAYER_SCALE);
7980

8081
} // namespace TNN_NS

0 commit comments

Comments
 (0)