colin
/
TIM-VX
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add macro VSI_EXPAND_BROADCAST_ENABLE_DIMENSIONS for ovxlib compatibility (#374)

This commit is contained in:
Antkillerfarm 2022-04-24 18:38:56 +08:00 committed by GitHub
parent dbb3631d4e
commit 3f2e67b65f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/tim/vx/internal/include/ops/vsi_nn_op_expand_broadcast.h
View File

@ -30,6 +30,8 @@
extern "C" { extern "C" {
#endif #endif
#define VSI_EXPAND_BROADCAST_ENABLE_DIMENSIONS
typedef struct _vsi_nn_expand_broadcast_param typedef struct _vsi_nn_expand_broadcast_param
{ {
uint32_t *shape; uint32_t *shape;

3
src/tim/vx/ops/broadcast.cc
View File

@ -37,6 +37,7 @@ Broadcast::Broadcast(Graph* graph, const std::vector<int32_t>& shape,
dimensions_(dimensions) { dimensions_(dimensions) {
this->impl()->node()->nn_param.expand_broadcast.dim_num = shape_.size(); this->impl()->node()->nn_param.expand_broadcast.dim_num = shape_.size();
this->impl()->node()->nn_param.expand_broadcast.shape = (uint32_t*)shape_.data(); this->impl()->node()->nn_param.expand_broadcast.shape = (uint32_t*)shape_.data();
#ifdef VSI_EXPAND_BROADCAST_ENABLE_DIMENSIONS
this->impl()->node()->nn_param.expand_broadcast.dimensions_num = dimensions_.size(); this->impl()->node()->nn_param.expand_broadcast.dimensions_num = dimensions_.size();
if (dimensions.size() > 0) if (dimensions.size() > 0)
{ {
@ -44,7 +45,7 @@ Broadcast::Broadcast(Graph* graph, const std::vector<int32_t>& shape,
} else { } else {
this->impl()->node()->nn_param.expand_broadcast.dimensions = nullptr; this->impl()->node()->nn_param.expand_broadcast.dimensions = nullptr;
} }
#endif
} }
std::shared_ptr<Operation> Broadcast::Clone( std::shared_ptr<Operation> Broadcast::Clone(